栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

题目描述:

 

 题解一:层序遍历 通过

1.对二叉树进行层序遍历(通过借助队列实现),将各个节点按照层序遍历结果保存在list中。

2.将每层层序遍历结果中除了最后一个节点的next设置为当前层下一个节点,最后一个节点的next设置为None。

class Solution(object):
    def connect(self, root):
        noderes = []
        layers = []
        if root==None:
            return root
        layer = []
        layer.append(root)
        noderes.append(layer)
        while noderes:
            nowlayer = noderes[0]
            noderes.remove(nowlayer)
            layers.append(nowlayer)
            newlayer = []
            for node in nowlayer:
                if node.left:
                    newlayer.append(node.left)
                if node.right:
                    newlayer.append(node.right)
            if len(newlayer)>0:
                noderes.append(newlayer)
        for eachlayer in layers:
            for i in range(len(eachlayer)-1):
                eachlayer[i].next = eachlayer[i+1]
            eachlayer[-1].next = None
        return root

 题解二:递归

参考https://blog.csdn.net/qq_32424059/article/details/90487297

1.如果root左右子节点都存在,left.next则为右子节点,right.next为root.next的子节点(左子节点如果存在则为左子节点,否则为右子节点,如果都不存在,则找到root.next.next,如果都不存在则填充为None)

2.因此创建一个函数findcousin(node,parent) ,node为parent的子节点,该函数试图在root的cousin节点中找到一个有子节点的,并将node.next设置为该cousin节点的左子节点(如果存在)或右子节点。

3.如果root 左右子节点均存在,left.next即为right,然后对root.right调用findcousin,如果只有left,则对left调用findcousin,如果只有right,则对right调用。

4.对root的左右子树调用connect,需要先对right进行处理,因为findcousin需要用到该层当前节点之后的信息。

class Solution(object):
    def connect(self, root):
        if root==None:
            return root
        def findcousin(node,parent):
            tmp = parent.next
            while tmp:
                if tmp.left:
                    node.next = tmp.left
                    break
                elif tmp.right:
                    node.next = tmp.right
                    break
                tmp = tmp.next
        if root.left and root.right:
            root.left.next = root.right
            findcousin(root.right,root)
        elif root.left:
            findcousin(root.left,root)
        elif root.right:
            findcousin(root.right,root)
        self.connect(root.right)
        self.connect(root.left)
        return root

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/618934.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号