leetcode 96:不同的二叉搜索树

使用动态规划的方式,

1 2 3 4 5 

用数组a来存储个数  

相当于是1为root  2 3 4 5形成一个数  所以是a[4]

2为root  前面1left 为a[1]  后面3 4 5为right a[3]  a[1]*a[3]

3为root 1 2left a[2]  4 5为right  a[2]   a[2]*a[2]

...

a[5]=a[4]*2+a[1]*a[3]*2 +a[2]*a[2]

 int numTrees(int n) {
    std::vector<int> a(n+1);
    a[0]=0;
    a[1]=1;
    a[2]=2;
    if(n<=2)
        return a[n];
    for(int i=3;i<n+1;i++){
        if(i%2==1) {
            for (int j = 1; j <= (i + 1) / 2; j++) {
                if (j == 1)
                    a[i] += a[i - 1] * 2;
                else if(j==(i+1)/2)
                    a[i] += a[i - j] * a[j - 1];
                else
                    a[i] += a[i - j] * a[j - 1]*2;
            }
        }
        else
        {
            for (int j = 1; j <= i/ 2; j++) {
                if (j == 1)
                    a[i] += a[i - 1] * 2;
                else
                    a[i] += a[i - j] * a[j - 1]*2;
            }
        }
    }
    return a[n];
}

猜你喜欢

转载自blog.csdn.net/u013263891/article/details/85227322