Populating Next Right Pointers in Each Node II

使用迭代,时间复杂度O(n),空间复杂度O(1)。逐层处理。代码如下:

class Solution {
public:
    void connect(TreeLinkNode *root) {
        TreeLinkNode *cur, *start, *next;
        start = root;
        while(start)
        {
            TreeLinkNode *p = NULL;
            next = NULL;  
            for(cur = start; cur; cur = cur->next)
            {
                if(cur->left)
                {
                    if(p) p->next = cur->left;;
                    p = cur->left;
                    if(!next)
                        next = cur->left;
                }
                if(cur->right)
                {
                    if(p) p->next = cur->right;
                    p = cur->right;
                    if(!next)
                        next = cur->right;
                }
            }
            start = next;
        }
    }
};

9.02 最新代码:

class Solution {
public:
    void connect(TreeLinkNode *root) {
        if(!root) return;
        
        TreeLinkNode *node = root;
        while(node)
        {
            TreeLinkNode *nextLevel = NULL, *prev = NULL;
            
            for(; node; node = node->next)
            {
                if(nextLevel == NULL)  nextLevel = (node->left)?(node->left):(node->right);
                
                if(node->left)
                {
                    if(prev) prev->next = node->left;
                    prev = node->left;
                }
                if(node->right)
                {
                    if(prev) prev->next = node->right;
                    prev = node->right;
                }
                
            }
            node = nextLevel;
        }
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值