Populating Next Right Pointers in Each Node I II@LeetCode

Populating Next Right Pointers in Each Node I

树的广度优先搜索题。记录下每一层的节点总个数，而后根据广度优先搜索的原则进行遍历，将非null节点都加入到队列中，对于同一层中的节点，将其next指向队列中的下一个节点便可。

实现代码：

javapublic class Solution {
    public void connect(TreeLinkNode root) {
        if (root == null)
            return;
        LinkedList<TreeLinkNode> nodes = new LinkedList<TreeLinkNode>();
        nodes.add(root);
        int numOfLevelTotal = 1;
        while (!nodes.isEmpty()) {
            TreeLinkNode treeLinkNode = nodes.poll();
            numOfLevelTotal--;
            if (treeLinkNode.left != null) {
                nodes.add(treeLinkNode.left);
            }
            if (treeLinkNode.right != null) {
                nodes.add(treeLinkNode.right);
            }
            if (numOfLevelTotal > 0) {
                treeLinkNode.next = nodes.getFirst();
            } else {
                numOfLevelTotal = nodes.size();
            }
        }
    }
}

Populating Next Right Pointers in Each Node II

根据题目来讲，这一题和上一次的区别在于：

What if the given tree could be any binary tree? Would your previous solution still work?

可是因为以前所采用的方法并无这种局限，因此直接拷贝过来也能够AC。

不过这里存在一个问题，仔细看题目里的要求：

You may only use constant extra space.

理论上，采用队列的话是确定没办法只是用常数额外内存的，可是LeetCode好像在这件事上没有检测的这么严，起码我写的Java代码和我看到的用递归解决的C++代码均可以经过。

那么若是硬要纠结一下这一条呢？不使用队列怎么作？

其实也不难，只是思路要转变一下，就不能是遍历这一层同时链接这一层，而是遍历这一层链接下一层。那么比较重要的就是要记录每一层的头节点，因为每一层在被遍历的时候是已经链接好了的，因此没必要担忧找不到节点的问题，若是挨个找寻next节点的子节点便可，子节点先内部（相同父节点）链接，而后再链接到总链表中便可。

实现代码：

javapublic class Solution {
    public void connect(TreeLinkNode root) {
        TreeLinkNode levelHead = root, nextLevelHead = null;
        while (levelHead != null) {
            TreeLinkNode node = levelHead, tail = null;
            while (node != null) {
                if (node.left != null && node.right != null) {
                    node.left.next = node.right;
                }
                TreeLinkNode sub;
                if (node.left != null)
                    sub = node.left;
                else if (node.right != null)
                    sub = node.right;
                else
                    sub = null;
                if (sub != null) {
                    if (nextLevelHead == null) {
                        nextLevelHead = sub;
                        tail = sub;
                    } else {
                        tail.next = sub;
                    }
                    while (tail.next != null)
                        tail = tail.next;
                }
                node = node.next;
            }
            levelHead = nextLevelHead;
            nextLevelHead = null;
        }
    }
}