263,填充每个节点的下一个右侧节点指针

给定一个满二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:

struct Node {

  int val;

  Node *left;

  Node *right;

  Node *next;

}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。

初始状态下,所有 next 指针都被设置为 NULL。

263,填充每个节点的下一个右侧节点指针

 

输入:

{"$id":"1","left":{"$id":"2","left":{"$id":"3","left":null,"next":null,"right":null,"val":4},"next":null,"right":{"$id":"4","left":null,"next":null,"right":null,"val":5},"val":2},"next":null,"right":{"$id":"5","left":{"$id":"6","left":null,"next":null,"right":null,"val":6},"next":null,"right":{"$id":"7","left":null,"next":null,"right":null,"val":7},"val":3},"val":1}

 

输出:

{"$id":"1","left":{"$id":"2","left":{"$id":"3","left":null,"next":{"$id":"4","left":null,"next":{"$id":"5","left":null,"next":{"$id":"6","left":null,"next":null,"right":null,"val":7},"right":null,"val":6},"right":null,"val":5},"right":null,"val":4},"next":{"$id":"7","left":{"$ref":"5"},"next":null,"right":{"$ref":"6"},"val":3},"right":{"$ref":"4"},"val":2},"next":null,"right":{"$ref":"7"},"val":1}

 

解释:给定二叉树如图 A 所示,你的函数应该填充它的每个next 指针,以指向其下一个右侧节点,如图 B 所示。

答案:

 1public Node connect5(Node root) {
2    dfs(root, null);
3    return root;
4}
5
6private void dfs(Node curr, Node next) {
7    if (curr == null) return;
8    curr.next = next;
9    dfs(curr.left, curr.right);
10    dfs(curr.right, curr.next == null ? null : curr.next.left);
11}

解析:

看到这题我们首先想到的是递归。代码很好理解,再来看个类似的解法

 1public Node connect3(Node root) {
2    if (root != null && root.left != null)
3        connectNodes(root.left, root.right);
4    return root;
5}
6
7public void connectNodes(Node node1, Node node2) {
8    node1.next = node2;
9    if (node1.left != null) {
10        connectNodes(node1.right, node2.left);
11        connectNodes(node1.left, node1.right);
12        connectNodes(node2.left, node2.right);
13    }
14}

注意题中说的是满二叉树,如果他的左子节点为空,那么他的右子节点也一定是为空的。下面再来看个非递归的解法

 1public Node connect1(Node root) {
2    if (root == null)
3        return null;
4    Node pre = root;
5    Node cur = null;
6    while (pre.left != null) {
7        cur = pre;
8        while (cur != null) {
9            cur.left.next = cur.right;
10            if (cur.next != null)
11                cur.right.next = cur.next.left;
12            cur = cur.next;
13        }
14        pre = pre.left;
15    }
16    return root;
17}

这种解法也比较容易理解,他会一层一层的往下遍历,只有把当前层连完之后才会操作下一层

上一篇:Mysql - 整数类型的存储字节数和范围


下一篇:LeetCode 263 丑数(easy)