Leetcode刷题笔记 96.不同的二叉搜索树

示例1
输入：
3

输出：
5

说明：

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

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

解法

n = 1
1
 
n = 2
     1          2
      \        /
       2      1
 
n = 3
    1            1           2                3      3
     \            \         / \              /      /
      2            3       1   3            2      1
       \          /                        /        \
        3        2                        1          2
  左边节点0个右边节点2个 左边节点1个右边节1一个  左边节点2个右边节点0个

dp[]代表有几个节点的时候有几种方法。

1. 左边节点或者右边节点个数为0的时候，dp[0]=1
2. 左边节点或者右边节点个数为1的时候, dp[1]=1
3. n=2时，有两种情况，
   1. 左边节点数为0，右边节点数=n-1-0（n-1，因为一个节点当根节点去了)，总共dp[0]*dp[1]=1
   2. 左边节点数为1，右边节点数=n-1-1，总共dp[1]*dp[0]=1
      两种情况相加dp[2]=2
4. n=3时，有三种情况，
   1. 左边节点数为0，右边节点数=n-1-0，总共dp[0]*dp[2]=2
   2. 左边节点数为1，右边节点数=n-1-1，总共dp[1]*dp[1]=1
   3. 左边节点数为2，右边节点数=n-1-2，总共dp[2]*dp[0]=2
      三种情况相加dp[3]=5
5. n=4时，有四种情况，
   1. 左边节点数为0，右边节点数=n-1-0，总共dp[0]*dp[3]=5
   2. 左边节点数为1，右边节点数=n-1-1，总共dp[1]*dp[2]=2
   3. 左边节点数为2，右边节点数=n-1-2，总共dp[2]*dp[1]=2
   4. 左边节点数为3，右边节点数=n-1-3，总共dp[3]*dp[0]=5
      四种情况相加dp[4]=15
6. 由此可以得到 卡特兰数 公式
   dp(n)=dp(0)∗dp(n−1)+dp(1)∗dp(n−2)+…+dp(n−1)∗dp(0)
7. 由此可以得出递推式
   dp(0) = 1 dp(n+1)=2*(2n+1)/(n+2)*dp(n)

代码

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
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=1;j<=i;j++){
                dp[i]+=dp[j-1]*dp[i-j];
            }
        }
        return dp[n];
    }
};
int main()
{
	//n=5 42 n=6 132
    int n = 5;
    Solution s ;
    cout<<s.numTrees(n);
}

今天也是爱zz的一天哦！