题目描述
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
解题思路
1、题意理解:根据二叉树的定义,
如果给定序列[1, n], 以i为根节点,那么[1, (i -1)]为根节点i的左子树,[i+1, n]为根节点的右子树
2、解题思路:动态规划
(1)**设G(n)为序列长度为n,不同二叉树的总数,
设F(i,n),i的取值范围为[1,n],序列长度为n,以i为根节点,不同二叉树的总数
G(n) = F(1, n) + F(2, n) + … + F(n, n) (a)
(2)根据题意,以i为根节点,左子树的结点数为i-1, 右子树的结点数为n -i
F(i, n) = G(i-1) * G(n-i) 1 <= i <= n (b)
(3)将(b)代入(a)可以得到:
G(n) = G(0) * G(n-1) + G(1) * G(n-2) + … + G(n-1) * G(0)
代码
class Solution {
public int numTrees(int n) {
int [] G = new int[n+1];
G[0] = G[1] = 1;
for(int i=2; i<= n; ++i){
for (int j=1; j<=i; ++j){
G[i] += G[j-1] * G[i -j];
}
}
return G[n];
}
}
参考:https://leetcode-cn.com/problems/unique-binary-search-trees/solution/hua-jie-suan-fa-96-bu-tong-de-er-cha-sou-suo-shu-b/
https://leetcode.com/problems/unique-binary-search-trees/discuss/31666/DP-Solution-in-6-lines-with-explanation.-F(i-n)-G(i-1)-*-G(n-i)