Leetcode.96 不同的二叉搜索树

题目链接

Leetcode.96 不同的二叉搜索树 mid

题目描述

给你一个整数 $n$ ，求恰由 $n$ 个节点组成且节点值从 $1$ 到 $n$ 互不相同的 二叉搜索树 有多少种？

返回满足题意的二叉搜索树的种数。

示例 1：

输入：n = 3
输出：5

示例 2：

输入：n = 1
输出：1

提示：

• $1<=n<=19$

解法：dp & 卡特兰数

我们定义 $g(n)$ 为 $n$ 个节点所能表示的不同的二叉搜索树的数量，这也就是我们最终要返回的答案。

我们定义 $f(i)$ 是以 节点$i$ 为根节点的二叉搜索树的数量。

那么，我们可以得出：

$$g(n)=f(1)+f(2)+f(3)+...+f(n-1)+f(n)$$

由于 $f(i)$ 是以 节点$i$ 为根节点，所以它的左子树中的节点数量为 $i-1$ ， 它的右子树中的节点数量为 $n-i$，所以：

$$f(i)=g(i-1)\times g(n-i)$$

讲以上两个式子结合一下，可以得出最终的式子：

$$g(n)=g(0)\times g(n-1)+g(1)\times g(n-2)+...+g(n-2)\times g(1)+g(n-1)\times g(0)$$

这个就是卡特兰数。

什么是卡特兰数

卡特兰数能够解决的问题

时间复杂度： $O(n^2)$

C++代码：

class Solution {
public:
    int numTrees(int n) {
        int g[n + 1];
        memset(g,0,sizeof g);

        g[0] = 1 , g[1] = 1;

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