Leecode 117. 填充每个节点的下一个右侧节点指针 II

题目：

给定一个二叉树

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}

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

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

示例：

输入：{“KaTeX parse error: Expected '}', got 'EOF' at end of input: …":"1","left":{"id”:“2”,“left”:{“KaTeX parse error: Expected 'EOF', got '}' at position 53: …t":null,"val":4}̲,"next":null,"r…id”:“4”,“left”:null,“next”:null,“right”:null,“val”:5},“val”:2},“next”:null,“right”:{“KaTeX parse error: Expected '}', got 'EOF' at end of input: …null,"right":{"id”:“6”,“left”:null,“next”:null,“right”:null,“val”:7},“val”:3},“val”:1}

输出：{“KaTeX parse error: Expected '}', got 'EOF' at end of input: …":"1","left":{"id”:“2”,“left”:{“KaTeX parse error: Expected '}', got 'EOF' at end of input: …:null,"next":{"id”:“4”,“left”:null,“next”:{“KaTeX parse error: Expected 'EOF', got '}' at position 53: …t":null,"val":7}̲,"right":null,"…id”:“6”,“left”:null,“next”:null,“right”:{“KaTeX parse error: Expected 'EOF', got '}' at position 9: ref":"5"}̲,"val":3},"righ…ref”:“4”},“val”:2},“next”:null,“right”:{"$ref":“6”},“val”:1}

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

提示：

你只能使用常量级额外空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

• 题目本身的解法的并不难，基本看一眼就知道广搜可以很简单的解决问题，但仔细看提示，要求常量级额外空间，这就限制了广搜的使用，再继续看提示，递归产生的栈空间占用不算空间复杂度，那意图就很明显了，要求我们用深搜来解决问题；
• 第一反应如果上一层已经连接好了，那么下一层肯定没问题，但是深搜做不到这一点；那么看一看前序遍历，每次遍历节点时，将左子节点（如果存在）指向右子节点，右子节点指向兄弟节点的左子节点或者右子节点（左子节点不存在时），看看是否能达成我们的目的，来模拟一下：
  遍历节点1：
  遍历节点2：
  遍历节点4：
  遍历节点5：
  
• 经过我们一轮模拟，看起来还不错，但上述例子的情况是每个节点都有子节点的情况，如果某个节点没有子节点会如何？假设我们拿掉节点11和12再来模拟一遍：
  遍历节点1：
  遍历节点2：
  
  遍历节点4：
  
• 到这里就不对了，按照题目要求，节点9应该指向10，但是按照我们的思路，遍历完节点4，我们就再也不会操作节点9的next引用了，遍历节点4时，因为节点5没有子节点，所以节点9被我们设置为null了；那么问题有两个：

1. 当处理节点4的右子节点9的next引用时，我们不能只看兄弟节点5是否有子节点，我们应该依次查看节点5，6，7是否有左或者右子节点，如果没有，我们才能将节点9的next设置为null；
2. 当我们按照刚才的遍历顺序，我们无法通过节点4访问到节点7，无法得知节点7是否存在子节点；

• 第一个问题好解决，当我们发现节点4的next节点（5）没有子节点时，就找5的next节点（6），直到找到某个节点有子节点或者所有跟节点4同级的节点（在次例子中为：节点5，6，7）都被确认没有子节点为止；
• 第二个问题在于，节点6的next没有指向节点7，但在我们的方法中，节点6的next最终会指向节点7，但是因为使用的是前序遍历，在遍历节点4时候，节点3还没有遍历，所以节点6的next暂时还没有指向节点7，那么我们修改一下遍历顺序即可，按中右左的顺序，遍历节点即可解决问题二；
• 所以最后的解决方法为：使用 中-右-左的顺序遍历二叉树，每次遍历节点时候如果节点有左子节点，那么需要操作左子节点next引用，如果该节点存在右子节点，那么next指向右子节点，如果不存在右子节点，则遍历处于该节点同级的且位于该节点右侧的节点，查找其是否存在子节点，next引用指向找到的第一个子节点或者null（都没有子节点）；

代码分析：

1. 中-右-左的顺序遍历二叉树；
2. 遍历节点时，操作子节点的next引用；

代码实现：

数组转二叉树：

/*
// Definition for a Node.
class Node {
    public int val;
    public Node left;
    public Node right;
    public Node next;

    public Node() {}

    public Node(int _val,Node _left,Node _right,Node _next) {
        val = _val;
        left = _left;
        right = _right;
        next = _next;
    }
};
*/
class Solution {
    public Node connect(Node root) {
        if(root == null) return root;
        preorder(root);
        return root;
    }
    private void preorder(Node node){
    //如果不存在子节点，那么该节点就无需操作
        if(node.left == null && node.right == null){
            return;
        }
        //存在左子节点
        if(node.left != null){
            if(node.right != null){
                node.left.next = node.right;
            } else if (node.next != null){//在同级，节点的右边存在别的节点
                Node temp = node.next;
                //查找下一个存在子节点的同级且位于右侧的节点
                while(temp != null && temp.left == null && temp.right == null) temp = temp.next;
                //找到符合要求的节点
                if(temp != null){
                    node.left.next = temp.left != null ? temp.left : temp.right;
                }
                //未找到符合要求的节点则不操作，因为默认next就是null
            }
        }
        //存在右子节点，操作与左子节点类似；
        if(node.right != null){
            if(node.next != null){
                Node temp = node.next;
                while(temp != null && temp.left == null && temp.right == null) temp = temp.next;
                if(temp != null){
                    node.right.next = temp.left != null ? temp.left : temp.right;
                }
            }
        }
        if(node.right != null)preorder(node.right);
        if(node.left != null) preorder(node.left);
    }
}