zl程序教程

您现在的位置是:首页 >  其他

当前栏目

【leetcode】114: 二叉树展开为链表

2023-09-11 14:21:08 时间

题目如下:

 

这个题目很有意思,就是我们可以先用先序遍历遍历整棵树,然后再重新进行原地修改,就可以了。代码如下:

# 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.
        """
        cur=root
        def dfs(root,ls):
            if root==None:
                return None
            ls.append(root.val) 
            dfs(root.left,ls)

            dfs(root.right,ls)
        ls=[]
        dfs(root,ls)
        for i in ls:
            print(i)

        i=0
        while i<len(ls):

            if i!=0:
                root.right=TreeNode(ls[i])
                root.left=None
                root=root.right
            if i==0:
                root.left==None
            i+=1

这样就可以了,有一个坑就是在进行原地修改的时候,一定要记得讲root.left进行删除,令其=none,不然难以得到正确的结果