弱问第95题Unique Binary Search Trees II

找到一个python代码

class Solution:
    def generateTrees(self, n):
        if n == 0:
            return []
        def generate(start, end):
            res = []
            for root in range(start, end + 1):
                leftTrees = generate(start, root - 1)
                rightTrees = generate(root + 1, end)
                for left in leftTrees:
                    for right in rightTrees:
                        node = TreeNode(root)
                        #print( node)
                        node.left = left
                        node.right = right
                        res.append(node)
            return res or [None]
        return generate(1, n)        

res把每个node append了，改成res.append(node.val)就不行了。看example

Input: 3
Output:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]

输出数值就可以了，为什么需要append一个node呢

想想res的类型是什么，node.val的类型又是什么？

usr_opta 发表于 2020-4-27 07:39
想想res的类型是什么，node.val的类型又是什么？

res的类型是list, node.val的类型是int。是说题目

generate all structurally unique BST's (binary search trees) that store values 1 ... n.

要的是BST，而不是像example里的

Output:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]

看上去是整数就行了？

对。而且题目已经写了要求返回值是List[TreeNode]类型。你往List里塞int就变成List[int]类型了。
你把语言切换成C++或者Java能更清楚看到类型要求。