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;
        }
    }
}