不同的二叉搜索树2

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

示例:

这道题只让计数，并不需要给出树结构。所以如果直接用计算树结构的方法会超时。可以发现n的计算依赖于前面的结果，所以可以用动态规划法。

设dp[i]为i个节点可以组成二叉搜索树的个数。且dp[0],dp[1] = 1。i = 2时，根节点固定，dp[2] = 左边0个节点个数 * 右边1个节点个数 + 左边1个节点个数 * 右边0个节点个数。依此类推…

class Solution {
public:

    int numTrees(int n) {
        vector<int> dp(n + 1, 0);
        dp[0] = 1;
        dp[1] = 1;

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

        return dp[n];
    }
};