Leetcode 96. 不同的二叉搜索树 C++

Leetcode 96. 不同的二叉搜索树

题目

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

测试样例

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

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

题解

动态规划
令dp[i]表示i个节点组成的二叉搜索树的数目
根据二叉搜索树的特点，左子树的节点都小于根节点，右子树的节点都大于根节点。因此构造的过程中，我们遍历每个点充当根节点，假设我们现在有i个节点，我们让j+1作为根节点，则小于j+1的j个节点构成的二叉搜索树便成为了左子树；j+2~i的i-j-1个节点构成的二叉搜索树便成为了右子树。二者组合便成了所有的二叉搜索树，显然，我们的动态转移方程为

• dp[i] += dp[j]*dp[i-j-1]，其中j>=0 && j<i
  初始条件就是1、2个节点时，都只有一种二叉搜索树，即dp[0]=dp[1]=1

代码

int numTrees(int n) {
        if(n == 0)  return 0;
        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-j-1];
            }
        }
        return dp[n];
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。