2020-11-17 114. 二叉树展开为链表

1、先完成了前序遍历:144. 二叉树的前序遍历
【根左右】

class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        if not root:
            return None
        nodelist=[]
        self.dfs(root,nodelist)
        return nodelist

    def dfs(self, root: TreeNode,nodelist):
        if not root: #叶子节点的左右为None,会直接返回
            return
        nodelist.append(root.val)
        self.dfs(root.left,nodelist)
        self.dfs(root.right,nodelist)

11-18前序遍历的堆栈法:

class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        res=[]
        if root is None:
            return res
        node=root
        stack=[]
        while node or stack:
            while node:
                res.append(node.val)
                stack.append(node)
                node=node.left #向左走
            node=stack.pop() #返回头节点
            node=node.right #取右边
        return res

2、把取出来的值,重构成树的结构。左子树为None,右子树为下一节点。注意size的范围(节点的是0~size-1, 叶子是1~size)。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        if not root:
            return None
        nodelist=[]
        self.dfs(root,nodelist)
        size=len(nodelist)
        size=size-1
        i=0
        while size:
            nod=nodelist[i]
            nod.left=None
            nod.right=nodelist[i+1]
            i=i+1
            size=size-1



    def dfs(self, root: TreeNode,nodelist):
        if not root:
            return
        nodelist.append(root)
        self.dfs(root.left,nodelist)
        self.dfs(root.right,nodelist)
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值