ACM学习历程—51NOD 1412 AVL树的种类(递推)

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <set>
#include <map>
#include <queue>
#include <vector>
#include <string>
#define LL long long
#define MOD 1000000007

using namespace std;

const int maxN = 2002;
LL p[maxN][maxN];

void init()
{
    int from, to, tfrom, tto;
    memset(p, 0, sizeof(p));
    p[0][0] = 1;
    p[1][1] = 1;
    from = 0; to = 1;
    for (int i = 2; i < maxN; ++i)
    {
        tfrom = to;
        tto = from;
        to++;
        for (int j = from; j <= to; ++j)
        {
            for (int k = 0; k < i; ++k)
            {
                p[i][j] += p[k][j-1]*p[i-1-k][j-1]%MOD;
                if (j > 1)
                {
                    p[i][j] += p[k][j-2]*p[i-1-k][j-1]%MOD;
                    p[i][j] += p[k][j-1]*p[i-1-k][j-2]%MOD;
                }
                p[i][j] %= MOD;
                if (p[i][j])
                {
                    tfrom = min(tfrom, j);
                    tto = max(tto, j);
                }
            }
        }
        from = tfrom;
        to = tto;
    }
    //cout << "OK"<<endl;
}

int main()
{
    //freopen("test.in", "r", stdin);
    init();
    int n;
    while (scanf("%d", &n) != EOF)
    {
        int ans;
        LL t = 0;
        for (int i = 1; i <= n; ++i) t = (t+p[n][i])%MOD;
        ans = t;
        printf("%d\n", ans);
    }
    return 0;
}