leetcode 96 不同的二叉搜索树
z
 2019-07-04 20:04:29  

给定一个整数 n，求以 1 … n 为节点组成的二叉搜索树有多少种？

示例:

输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:

1
2
3
4
5
1         3     3      2      1
  \       /     /      / \      \
   3     2     1      1   3      2
  /     /       \                 \
 2     1         2                 3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Solution(object):
    def numTrees(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n == 1:
            return 1
        if n == 2:
            return 2
        if n == 3:
            return 5
        dp = [0 for i in range(n)]
        dp[0] = 1
        dp[1] = 2
        dp[2] = 5
        for i in range(3, n):
            for j in range(i+1):
                if j == 0:
                    dp[i] += dp[i-1]
                elif j==i:
                    dp[i] += dp[i-1]
                else:
                    dp[i] += dp[j-1] * dp[i-j-1]
        return dp[-1]