leetcode 117. 填充每个节点的下一个右侧节点指针 II

踩着七彩祥云的猴子

于 2019-12-15 15:37:06 发布

阅读量79

点赞数

分类专栏： python leetcode

本文链接：https://blog.csdn.net/weixin_36149892/article/details/103549277

版权

leetcode 同时被 2 个专栏收录

148 篇文章 0 订阅

订阅专栏

python

62 篇文章 0 订阅

订阅专栏

# 117. 填充每个节点的下一个右侧节点指针 II
# https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii
from copy import copy


class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if root is None:
            return root
        cur, nxt = [], []
        cur.append(root)
        root.next = None

        while cur:
            while cur:
                tmp = cur.pop(0)
                if tmp.left:
                    nxt.append(tmp.left)
                if tmp.right:
                    nxt.append(tmp.right)
            if len(nxt) != 0:
                for i in range(len(nxt) - 1):
                    nxt[i].next = nxt[i + 1]
                nxt[len(nxt) - 1].next = None
            # 这里如果用deepcopy的话，你会爽歪歪！！！ 你要用原来的树节点，而不是deepcopy创建的新节点
            cur = copy(nxt)
            nxt.clear()
        return root