116. 填充每个节点的下一个右侧节点指针_117. 填充每个节点的下一个右侧节点指针 II

116. 填充每个节点的下一个右侧节点指针

问题

给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

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

例子

思路
完美二叉树：除了叶子结点之外的每一个结点都有两个孩子结点

//方法1
class Solution {
    public Node connect(Node root) {
        if(root==null) return root;
        Queue<Node> q = new LinkedList<>();
        q.offer(root);
        Node pre=null,node=null;
        while(q.size()>0) {
            int len = q.size();
            //获取该层第一个结点pre，.next=node
            pre = q.poll();
            if(pre.left!=null) q.offer(pre.left);
            if(pre.right!=null) q.offer(pre.right);
            for(int i=1; i<len; i++) {
                node =q.poll();
                if(node.left!=null) q.offer(node.left);
                if(node.right!=null) q.offer(node.right);
                
                pre.next = node;
                pre=node;
            }
            //最后一个结点.next=null
            pre.next=null;
        }
        return root;
    }
}
//方法2 递归
   if(root==null) return null;
   //是否有子结点,因为是完美二叉树,要么有两个子结点要么没有子结点
    //没有子节点
    if(root.left==null) return root;
     //有子节点
    root.left.next = root.right;
    if(root.next!=null)
        root.right.next = root.next.left;
    
    //root.left 下的节点的连接须要root.right下的节点联结成功

    connect(root.right);
    connect(root.left);
    return root;

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

问题
不是完美二叉树
代码

class Solution {
    public Node get(Node node) {
        while(node!=null) {
            if(node.left!=null) return node.left;
            if(node.right!=null) return node.right;
            node=node.next;
        }
        return null;
    }
    public Node connect(Node root) {
        if(root==null) return null;
        //如果有左子节点
        if(root.left!=null) {
            if(root.right!=null)
                root.left.next=root.right;
            else
                root.left.next=get(root.next);
        }
            
        // 如果有右子节点
        if(root.right!=null) {
            root.right.next=get(root.next);

        }
   	    //  root.left 下的节点的连接须要root.right下的节点联结成功

        //                  o root
        //                 / \
        //     root.left  o —— o  root.right
        //               /    / \
        //              o —— o   o
        //             /        / \
        //            o        o   o
        connect(root.right);
        connect(root.left);
        return root;
        
    }
}