LeetCode: Populating Next Right Pointers in Each Node II

最新推荐文章于 2021-09-24 12:05:40 发布

AIvin24

最新推荐文章于 2021-09-24 12:05:40 发布

阅读量572

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/AIvin24/article/details/38942119

版权

LeetCode 专栏收录该内容

144 篇文章 0 订阅

订阅专栏

思路：这个题目参考了这篇博客的思路，在每一层中，先用一个指针 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;
        }
    }
};