LeetCode96-不同的二叉搜索树

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

示例:

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

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3

一、思路

由二叉搜索树的中序遍历的性质，对任意的二叉搜索树的中序遍历为：$[1,2,...,n]$

这里，n个点都可以作为根结点，根据根结点，又可以清晰地找出左右子树，设根结点为$i$

则左子树为：$[1,2,...,i-1]$

右子树为：$[i+1,...,n]$

于是以$i$节点为根结点的二叉搜索树的数量为：左子树数量*右子树数量

又：左右子树也是二叉搜索树，于是可以通过递归算法计算以第$i$个节点为根结点的二叉搜索树的数量。

C++代码：

class Solution {
public:
	int numTrees(int n) {
		// dp[i][j]：以i起始，j作为结束的二叉搜索树的数量
		vector<vector<int>> dp(n + 1, vector<int>(n + 1, 0));
		for (int i = 1; i < n + 1; i++)
			dp[i][i] = 1;

		for (int j = 2; j < n + 1; j++)
			for (int i = j - 1; i > 0; i--) {
				for (int root = i; root <= j; root++) {
					if (root + 1 > j)
						dp[i][j] += dp[i][root - 1];
					else if (i > root - 1)
						dp[i][j] += dp[root + 1][j];
					else
						dp[i][j] += dp[i][root - 1] * dp[root + 1][j];
				}
			}
		return dp[1][n];
	}
};

执行效率：