原题:
Unique Binary Search TreesGiven n, how many structurally unique BST’s (binary search trees) that store values 1…n?
For example,
Given n = 3, there are a total of 5 unique BST’s.
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
这是一个动态规划问题:先从小问题开始分析,然后逐渐完成所有问题。
当节点小于等于1的时候,只有一种情况,返回1。
当有2个节点的时候,有两种情况,返回2.
然后从三个节点开始来求解,有三个节点时,根节点一共有两个子节点,可能的情况是1:两个左孩子 2:两个右孩子 3:一左一右
使用F(n)表示n个节点的不同二叉树数目,这3种情况可以表示为
1:F(2)* F(0)
2: F(0) * F(2)
3:F(1) * F(1)
这样就根据已知的F(0),F(1)和F(2)就求得了F(3)
同样利用F(0)–F(3)可以得到F(4)
当有4个节点的时候根节点有3个子节点,左右孩子的个数所有情况为(0,3)(1,2)(2,1)(3,0)
不同的二叉树个数为F(0)*F(3) F(1)*F(2) F(2)*F(1) F(3)*F(0)
根据这个方式可以迭代求的F(n)
class Solution {
public:
int numTrees(int n) {
if(n <= 1)
return 1;
else if(2 == n)
return 2;
vector<int> treeStruct;
treeStruct.push_back(1);
treeStruct.push_back(1);
treeStruct.push_back(2);
for(int i = 3; i <= n; ++i)
{
int sum = 0;
for(int j = 0; j < i; ++j)
{
sum += treeStruct[j] * treeStruct[i - 1 - j];
}
treeStruct.push_back(sum);
}
return treeStruct[n];
}
};