Python将一个数组变成树

问题

leetcode897遇到这样一个问题,将一个数组变成一个递增顺序搜索树,使用如下方法是错误的:

for i in tree_shuzu[::-1]:
        res.val = i
        res.right = res
print(res)

会提示:Error - Found cycle in the TreeNode

解决方案

参考题解后一种可行的方案如下:

temp = res
for i in tree_shuzu:
    res.right = TreeNode()
    res.right.val = i
    res = res.right
return (temp.right)

两个要点:
1、用temp存储最开始的树,后续对树再进行变动,返回temp就可以返回一整棵树
2、树的右节点要重新实例化,不然不能给右节点赋值

该题的完整代码如下:

# 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 increasingBST(self, root: TreeNode) -> TreeNode:
        tree_shuzu = []
        res = TreeNode()
        def digui(root):
            if root != None:
                digui(root.left)
                tree_shuzu.append(root.val)
                digui(root.right)
        
        digui(root)
        temp = res
        for i in tree_shuzu:
            res.right = TreeNode()
            res.right.val = i
            res = res.right
        return (temp.right)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值