117.Populating Next Right Pointers in Each Node II

题目链接

题目大意：与116题类似，区别是这题的二叉树是普通二叉树，而不是完全二叉树。要求空间复杂度o(1)。例子如下：

法一：直接用116题的层序遍历的办法。代码如下（耗时5ms）：

    public void connect(TreeLinkNode root) {
        if(root == null) {
            return;
        }
        Queue<TreeLinkNode> q = new LinkedList<TreeLinkNode>();
        q.offer(root);
        while(!q.isEmpty()) {
            //当前层一共有多少结点数
            int len = q.size();
            //遍历当前层的所有结点
            for(int i = 0; i < len; i++) {
                TreeLinkNode tmp = q.poll();
                //注意只计算到当前层倒数第二个结点即可。因为如果计算到最后一个的话，q中有值，会导致next出错。
                if(i < len - 1) {
                    tmp.next = q.peek();
                }
                if(tmp.left != null) {
                    q.offer(tmp.left);
                }
                if(tmp.right != null) {
                    q.offer(tmp.right);
                }
            }
        }
    }

法二：模仿116题的o(1)的解法，只是更灵活些，这里相当于直接给每一层形成了真正的一个单链表。具体代码如下（耗时1ms）：

    public void connect(TreeLinkNode root) {
        TreeLinkNode nextLevel = new TreeLinkNode(0), cur = nextLevel;
        while(root != null) {
            if(root.left != null) {
                //这里其实就是改变了pre.next，因为cur=pre，改变cur就是改变pre
                cur.next = root.left;
                cur = cur.next;
            }
            if(root.right != null) {
                cur.next = root.right;
                cur = cur.next;
            }
            root = root.next;
            //如果到了当前层结尾
            if(root == null) {
                //重置cur
                cur = nextLevel;
                //重置root为下一层的第一个结点
                root = nextLevel.next;
                //重置nextLevel
                nextLevel.next = null;
            }
        }
    }