LeetCode第 96 题：不同的二叉搜索树(C++)

96. 不同的二叉搜索树 - 力扣（LeetCode）

感觉更像是一个公式推导题，具体就直接看题解吧，推导挺简单的：不同的二叉搜索树 - 不同的二叉搜索树 - 力扣（LeetCode）

动态规划

一个重点：注意到 G(n) 代表长为i的序列能构成的二叉搜索树种数，它和序列的内容无关，只和序列的长度有关。

class Solution {
public:
    int numTrees(int n) {
        vector<int> G(n+1, 0);
        G[0] = 1;
        G[1] = 1; //G[i]代表长为i的序列能构成的二叉搜索树种数
        for(int i = 2; i <= n; ++i){//i代表序列长度
            for(int j = 1; j <= i; ++j){//j代表以j为根节点
                G[i] += G[j-1]*G[i-j];
            }
        }
        return G[n];
    }
};

如果没想到上述的推导的话，可以直接用递归，理解起来也容易，不同的二叉搜索树 - 力扣（LeetCode）

$$f(n)=f(0)\ast f(n-1)+f(1)\ast f(n-2)+...+f(n-1)\ast f(0)$$

递归

class Solution {
public:
    int numTrees(int n) {
        if(n == 0) return 1;
        if(n <3) return n;
        int sum = 0;
        for(int i = 0; i < n; ++i)
            sum += numTrees(i) * numTrees(n-i-1);
        return sum;
    }
};

上述递归的效率并不高，因为存在着大量的重复运算，所以明显就是要用记忆化递归了：

class Solution {
public:
    vector<int> vec;
    int numTrees(int n) {
        if(vec.size() == 0){
            vec.resize(30);
        }
        if(n == 0) return 1;
        if(n < 3) return n;
        int sum = 0;
        for(int i = 0; i < n; ++i){
            if(vec[i] == 0)
                vec[i] = numTrees(i);
            if(vec[n-i-1] == 0)
                vec[n-i-1] = numTrees(n-i-1);
            sum += vec[i]*vec[n-i-1];
        }
        return sum;
    }
};

或者这么写：

class Solution {
public:
    vector<int> vec;
    int numTrees(int n) {
        if(vec.size() == 0){
            vec.resize(n+1, 0);
            vec[0] = 1;
        }
        if(vec[n] != 0) return vec[n];
        int sum = 0;
        for(int i = 0; i < n; ++i)
            sum += numTrees(i) * numTrees(n-i-1);
        vec[n] = sum;
        return sum;
    }
};