树相关

目录

144前序遍历

思路:(循环前入栈、先右节点入栈)

  • 建栈,入栈,循环,只要栈不为空
    • 出栈,把值加入res。
    • 如果右不为空,入栈。左一样。
List<Integer> res = new ArrayList<>();
if (root == null) {
    return res;
}

Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
    TreeNode curNode = stack.pop();
    res.add(curNode.val);
    // 下面用于非递归的反转二叉树
//    TreeNode tempNode = node.left;
//    node.left = node.right;
//    node.right = tempNode;

    if (curNode.right != null) {
        stack.push(curNode.right);
    }
    if (curNode.left != null) {
        stack.push(curNode.left);
    }
}
return res;

94中序遍历(98验证二叉搜索树、230二叉搜索树中第K小的元素)

思路:

  • 建栈、cur指针,不入栈循环,只要cur和栈不为空
    • 只要cur不为空,循环让左子节点入栈,cur做相应移动
    • 出栈,把值加入res。
    • cur移动到右节点
// 设置计数器(二叉搜索树中第K小的元素)
// int cnt = 0;

List<Integer> res = new ArrayList<>();

if (root == null) {
    return res;
}

Stack<TreeNode> stack = new Stack<>();
TreeNode cur = root;
while (cur != null || !stack.isEmpty()) {

    // 先把左子节点都入栈
    /** 注意是cur != null,而不是cur.left != null */
    while (cur != null) {
        stack.push(cur);
        cur = cur.left;
    }

    cur = stack.pop();
    // (二叉搜索树中第K小的元素)
//      cnt++;
//      if (cnt == k) return cur.val;

    // (验证二叉搜索树)
//      if (pre != null && cur.val <= pre.val) return false;
//      pre = cur;
    res.add(cur.val);
    cur = cur.right;
}
return res;

145后序遍历

思路:

  • 建栈、cur和pre指针
  • pop依然在中间,但add前要判断是否还有右节点或者之前就是右节点,否则把cur放回去,指向它的右节点。add后pre成为cur,cur变为null
Stack<TreeNode> stack = new Stack<>();
TreeNode pre = null;
TreeNode cur = root;

while (cur != null || !stack.isEmpty()) {
    while (cur != null) {
        stack.push(cur);
        cur = cur.left;
    }

    cur = stack.pop();
    if (cur.right == null || pre == cur.right) {
        res.add(cur.val);
        pre = cur;
        cur = null;
    } else {
        stack.push(cur);
        cur = cur.right;
    }
}

102/107层次遍历(104二叉树最大深度、103

二叉树的锯齿形层次遍历)

思路:

  • 新建queue,入列
  • 循环,只要q不为空
    • 新建level链表,记录本层元素的个数
    • 遍历此层
      • 出列,加入res。
      • 如果左节点不为空,入列。右节点一样。
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);

while (!queue.isEmpty()) {
    // 下面用于 maxDepth
//      res++;

List<Integer> level = new ArrayList<>();
// 注意,由于循环中对queue进行修改,其size不断变化,所以这个值不能直接放到循环条件中
int length = queue.size();

// 下面用于 zigzagLevelOrder
//  boolean flag = result.size() % 2 == 0;

for (int i = 0; i < length; i++) { // 遍历一层的node
    TreeNode node = queue.poll();
    level.add(node.val);

    // 下面用于 zigzagLevelOrder
//      if (flag) level.add(node.val);
//      else level.add(0, node.val);

    if (node.left != null) queue.offer(node.left);
    if (node.right != null) queue.offer(node.right);
}
// levelOrderBottom就 add(0, level)
// 不断往0插入数据,旧数据就被挤到后面了
result.add(level);

105从前序与中序遍历序列构造二叉树

思路:

  • 新建preStart, preEnd, inStart, inEnd四个变量
  • 调用递归函数
    • 当“前序”或者“中序”中其中一个结束,就返回null
    • 取出pre中的preStart元素,新建节点,然后在中序数组中找出该节点的index。根据这个index来划分左右子树对应“前序”和“中序”的边界
// construct函数
if (preStart > preEnd || inStart > inEnd) {
    return null;
}

int val = preorder[preStart];
TreeNode p = new TreeNode(val);

int k = 0;
for (int i = 0; i < inorder.length; i++) {
    if (val == inorder[i]) {
        k = i;
        break;
    }
}

// 注意要减去inStart
p.left = construct(preorder, preStart + 1, preStart + (k - inStart),
        inorder, inStart, k - 1);
p.right = construct(preorder, preStart + (k - inStart) + 1, preEnd,
        inorder, k + 1, inEnd);

return p;

114二叉树展开为链表

    1
   / \
  2   5
 / \   \
3   4   6

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

只能说,从上面的观察结果来看,可以先处理右节点,再到左节点的顺序。

private TreeNode preNode = null;

public void flatten(TreeNode root) {
   if (root == null) {
        return;
    }
    flatten(root.right);
    flatten(root.left);
    root.right = preNode;
    root.left = null;
    preNode = root;
}

124二叉树中的最大路径和

思路:递归

  • 对左右节点递归调用函数,当节点为null时,返回0。如果返回的结果比0还小,可以不选该节点,所以以left为例,赋值为返回值及0中的最大值。
  • 返回左右节点值后,判断res是否需要更新,比较值是左右即当前节点值的和。
  • 最后只能返回一条路近的值,即从左到当前节点还是从右到当前节点的和。

235/236二叉树的最近公共祖先

236思路:

target是p和q的最近公共父节点,它有两种情况
要么p和q都不是target,要么p或q为target
第一种情况:p和q会分布在target的左右两边,所以左右返回的都不是null,返回当前即可
第二种情况:左右其中一个返回null,一个非null,直接返回非null即可
递归寻找,root绝对会等于null或者p、q中的一个

if (root == null || root == p || root == q) return root;

TreeNode left = lowestCommonAncestor1(root.left, p, q);
// 下面优化,说明左节点是一个非p或q的公共节点
// if (left != null && left != p && left != q) return left;
TreeNode right = lowestCommonAncestor1(root.right, p, q);

if (left != null && right != null) return root;

return left == null ? right : left;

235思路:

如果当前节点的值大于两个参数节点,说明它们的公共父节点在左边,反之。只有当当前节点的值处于两个值之间,当前节点才是公共父节点。

if (root.val > p.val && root.val > q.val) {
    return lowestCommonAncestor2(root.left, p, q);
} else if (root.val < p.val && root.val < q.val) {
    return lowestCommonAncestor2(root.right, p, q);
} else {
    return root;
}

猜你喜欢

转载自www.cnblogs.com/code2one/p/10100189.html
今日推荐