代码题--C++--不同的二叉搜索树

代码题--C++--不同的二叉搜索树

---

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

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

解题思路：二叉搜索树是指一个二叉树的中序遍历是递增的序列。假设n个节点存在二叉排序树的个数是G(n)，1为根节点，2为根节点，…，n为根节点。

就跟斐波那契数列一样，我们把n = 0 时赋为1，因为空树也算一种二叉搜索树，那么n = 1时的情况可以看做是其左子树个数乘以右子树的个数，左右字数都是空树，所以1乘1还是1。那么n = 2时，由于1和2都可以为跟，分别算出来，再把它们加起来即可。n = 2的情况可由下面式子算出：

dp[2] =  dp[0] * dp[1]　　　(1为根的情况)

　　　　+ dp[1] * dp[0]　　  (2为根的情况)

同理可写出 n = 3 的计算方法：

dp[3] =  dp[0] * dp[2]　　　(1为根的情况)

　　　　+ dp[1] * dp[1]　　  (2为根的情况)

 　　　  + dp[2] * dp[0]　　  (3为根的情况)

由此可以得出卡塔兰数列的递推式为：

 

 当1为根节点时，其左子树节点个数为0，右子树节点个数为n-1，同理当2为根节点时，其左子树节点个数为1，右子树节点为n-2，所以可得G(n) = G(0)G(n-1)+G(1)(n-2)+…+G(n-1)*G(0)。

代码如下：

class Solution {
public:
	int numTrees(int n) {
		vector<int> dp(n + 1, 0);
		dp[0] = 1;
		//利用G(n) = G(0)*G(n-1)+G(1)*(n-2)+...+G(n-1)*G(0)进行动态规划
		for (int i = 1; i <= n; ++i){
			for (int j = 0; j < i; ++j){
				dp[i] += (dp[j] * dp[i - 1 - j]);
			}
		}
		return dp[n];
	}
};