LeetCode: Populating Next Right Pointers in Each Node II

思路：这个题目参考了这篇博客的思路，在每一层中，先用一个指针 P 记录需要连接的节点，碰到下一个节点 Q 后，令 P->next = Q，而且更新P = Q，注意这是P不为空的情况，在每一层中，可以依次遍历每个节点，当一个节点的左孩子或者右孩子非空时，这这个孩子节点是下一层的 P的 初始值。这样就完成了空间复杂度为O(1)的实现。

code:

class Solution {
public:
    void connect(TreeLinkNode *root) {
        while(root){
            TreeLinkNode * nextLayer = NULL;
            TreeLinkNode * needToLink = NULL;
            TreeLinkNode * p = root;
            while(p && nextLayer == NULL){
                    nextLayer = p->left != NULL ? p->left : p->right;
                    p = p->next;
            }
            needToLink = nextLayer;
            while(root){
                if(root->left != NULL){
                    if(needToLink != root->left) needToLink->next = root->left;
                    needToLink = root->left;
                }
                if(root->right != NULL){
                    if(needToLink != root->right) needToLink->next = root->right;
                    needToLink = root->right;
                }
                root = root->next;
            }
            root = nextLayer;
        }
    }
};