51nod 1371 DP

有n行格子，第i(1<=i<=n)行有i个格子，每行格子是左对齐。现在要在每一个格子填入一个非负整数，最后使得每一行每一列的和都不超过2。

请计算有多少种方案，答案比较大，请输出对100,000,007(1e8+7)取余后的结果。

下图是n=4的时候格子的摆放。

Input

第1行：给出一个整数n (1<=n<=200)。

Output

对于每个测试用例，输出结果占一行。

Input示例

1

Output示例

3

官方题解：

DP题，状态dp[i][j][k]表示第i行填完之后有j列的和是1,k列的和是2。

然后加入第i+1行。

对第i+1行的最后一个格子进行分类讨论。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll __int64
#define mod 100000007
ll dp[205][205][205];//0的个数 1的个数
//

ll C[210][210];

void init()
{
    C[0][0]=1ll;
    C[1][0]=C[1][1]=1ll;
    for(int i=2;i<=205;i++)
    {
        C[i][i]=C[i][0]=1ll;
        for(int j=1;j<i;j++)
        {
            C[i][j]=C[i-1][j-1]+C[i-1][j];
            C[i][j]%=mod;
        }
    }
}

int main()
{
    int n;
//    freopen("in.txt","r",stdin);
    init();
    while(scanf("%d",&n)!=EOF)
    {

    memset(dp,0,sizeof(dp));
    dp[1][0][0]=dp[1][1][0]=dp[1][0][1]=1ll;
    for(int i=1;i<n;i++)
    {
       for(int j=0;j<=i;j++)
       {
           for(int k=0;k<=(i-j);k++)
           {
               int col0=j;
               int col1=k;
               col0++;

               for(int k1=0;k1<=min(col0,2);k1++)//放几个1在0
               {
//                   cout<<"m "<<j<<" "<<k<<endl;
                   for(int k2=0;k2<=min(col0-k1,1);k2++)//放几个2
                   {
                       for(int k3=0;k3<=col1;k3++)//放几个1在1
                       {
//                           cout<<"r"<<endl;
                           if(k1+k3>2) break;
                           if(col0-k2+col1-k3>i+1) break;
                           if(k1+k3+2*k2>2) continue;
                           int t1=col1-k3+k1;
                           int t2=col0-k2-k1;
                           int t3=i+1-t1-t2;
//                           if(t1==0 && t2==0 ) cout<<t1<<" "<<t2<<" "<<t3<<endl;
//                           if(2*t3+t2>2) continue;
                           dp[i+1][col0-k2-k1][col1-k3+k1]+=dp[i][col0-1][col1]*C[col0][k1]%mod*C[col0][k2]%mod*C[col1][k3]%mod;
//                           if(t1==0 && t2==0) cout<<(col0-k2-k1)<<" "<<(col1-k3+k1)<<" "<<dp[i+1][col0-k2-k1][col1-k3+k1]<<" "<<(col0-1)<<" "<<col1<<" "<<dp[i][col0-1][col1]<<endl;
                           dp[i+1][col0-k2-k1][col1-k3+k1]%=mod;
                       }
                   }
               }
           }
       }
    }

    ll ans=0;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=(n-i);j++)
        ans+=dp[n][i][j],ans%=mod;//cout<<i<<" "<<j<<" "<<dp[n][i][j]<<endl;
    printf("%I64d\n",ans);
    }
    return 0;

}