题目:给出一个数字n,假设用一棵搜索树来存储1~n,问这样的搜索树有多少个?例如,当n=3时,有5个独立的搜索树。
思路:有数字1,2,3,…,m,…n-1,n-2,那么当数字m作为搜索树的根时,左边的数字必须在它的左子树上。m的左边有m-1个数字,显然这是一个1~m-1的搜索树个数问题。m的右边有n-m个数,虽然数字是从m+1到n,但是实际上等同于1~n-m的搜索树个数问题。有个这个基本思路,就可以写代码了:
int numTrees(int n)
{
if(n <= 0)
return 0;
else if(n == 1)
return 1;
else
{
int tmp = 0;
for(int i = 2; i < n ; ++i)
tmp += numTrees(i-1) * numTrees(n-i);
tmp += 2*numTrees(n-1);
return tmp;
}
}
当然在无需考虑时间消耗和空间消耗的情况下可以这么做,但是这确实是一个糟糕的算法。在递归的时候多次调用同一个前一步已经计算出来的值,for循环里面加法步骤是对称的,例如numTrees(2)就会调用两次。应该避免这种情况,假设1~n的独立搜索树的个数表示为s(n),s(n)和s(n-1),s(n-2),…,s(2),s(1)有关,可以采用正推的办法,计算1,2,3,…,n-1,n时的s(n),同时保存每步的计算结果。
int numTrees(int n)
{
if(n <= 0)
return 0;
else
{
int* result = new int[n+1]();
result[0] = 1;
result[1] = 1;
for(int i = 2; i < n + 1; ++i)
{
for(int j = 1; j <= i; ++j)
result[i] += result[j-1] * result[i-j];
}
int tmp = result[n];
delete[] result;
return tmp;
}
}