LeeCode 96 DP

题意

传送门 LeeCode 96. 不同的二叉搜索树

题解

对于二叉搜索树，根节点的左右子树节点取值是连续有序的，可以观察到二叉搜索树的种数只与节点数量有关。$dp[i]$ 代表有 $i$ 个节点的二叉搜索树种数

$$dp[i]=\sum _{1\le j\le i}dp[j-1]\ast dp[i-j]$$

枚举根节点，记忆化搜索即可。

class Solution
{
#define maxn 100005
public:
    int dp[maxn];
    int rec(int n)
    {
        if (dp[n] != -1) return dp[n];
        if (n < 2) return dp[n] = 1;
        int res = 0;
        for (int i = 1; i <= n; i++)
        {
            res += rec(i - 1) * rec(n - i);
        }
        return dp[n] = res;
    }
    int numTrees(int n)
    {
        memset(dp, -1, sizeof(dp));
        return rec(n);
    }
};