leetcode_95. 不同的二叉搜索树 II

举报
悲恋花丶无心之人 发表于 2021/02/03 00:35:11 2021/02/03
【摘要】 目录 一、题目内容 二、解题思路 三、代码 一、题目内容 给定一个整数 n,生成所有由 1 ... n 为节点所组成的 二叉搜索树 。 示例: 输入:3 输出: [   [1,null,3,2],   [3,2,null,1],   [3,1,null,null,2], &...

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给定一个整数 n,生成所有由 1 ... n 为节点所组成的 二叉搜索树 。

示例:

输入:3
输出:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
解释:
以上的输出对应以下 5 种不同结构的二叉搜索树:

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3

 

提示:

0 <= n <= 8

二、解题思路

如果将 i 作为根节点,那么 [1, i) 为的左子树节点,(i, n] n为右子树节点。

那么我们需要求左子树 [1,i)的所有排列和右子树(i, n]的所有排列。

三、代码


  
  1. # Definition for a binary tree node.
  2. class TreeNode:
  3. def __init__(self, val=0, left=None, right=None):
  4. self.val = val
  5. self.left = left
  6. self.right = right
  7. def __repr__(self):
  8. return str(self.val)
  9. class Solution:
  10. def generateTrees(self, n: int) -> list:
  11. res = []
  12. if n < 1:
  13. return res
  14. def dfs(st, end):
  15. res = []
  16. if st > end:
  17. res.append(None)
  18. return res
  19. else:
  20. for i in range(st, end + 1):
  21. left = dfs(st, i - 1)
  22. right = dfs(i + 1, end)
  23. for l in left:
  24. for r in right:
  25. node = TreeNode(i)
  26. node.left = l
  27. node.right = r
  28. res.append(node)
  29. return res
  30. return dfs(1, n)
  31. if __name__ == '__main__':
  32. n = 3
  33. s = Solution()
  34. ans = s.generateTrees(n)
  35. print(ans)

文章来源: nickhuang1996.blog.csdn.net,作者:悲恋花丶无心之人,版权归原作者所有,如需转载,请联系作者。

原文链接:nickhuang1996.blog.csdn.net/article/details/108715933

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。