51node1371 填数字(dp)

51node1371

题目

中文题目

思路

dp[i][j][k]表示第i行填完之后有j列的和是1,k列的和是2，我们枚举每行的最后一个为0，1，2时的情况分类讨论。具体看代码，注意注释是有层次的。
（感觉状态设计得很好，学到了）

代码

#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#include <cstdio>

using namespace std;

typedef long long ll;

const int mod=1e8+7;

ll dp[205][205][205];
int n;

int main()
{
    scanf("%d",&n);
    dp[1][0][0]=dp[1][0][1]=dp[1][1][0]=1;
    for(int i=2; i<=n; i++)
    {
        for(int j=0; j<=i; j++)
            for(int k=0; j+k<=i; k++)
            {
                //末0
                    //加2
                if(i-j-k-1>=1)
                    dp[i][j][k+1]=(dp[i][j][k+1]+dp[i-1][j][k]*(i-j-k-1))%mod;
                    //加两个1
                        //加到00
                if(i-k-j-1>=2)
                    dp[i][j+2][k]=(dp[i][j+2][k]+dp[i-1][j][k]*((i-j-k-1)*(i-j-k-2)/2))%mod;
                        //加到01
                if(j>=1&&i-j-k-1>=1)
                    dp[i][j][k+1]=(dp[i][j][k+1]+dp[i-1][j][k]*j*(i-j-k-1))%mod;
                        //加到11
                if(j>=2)
                    dp[i][j-2][k+2]=(dp[i][j-2][k+2]+dp[i-1][j][k]*(j*(j-1)/2))%mod;
                    //加一个1
                        //加到1
                if(j>=1)
                    dp[i][j-1][k+1]=(dp[i][j-1][k+1]+dp[i-1][j][k]*j)%mod;
                        //加到0
                if(i-j-k-1>=1)
                    dp[i][j+1][k]=(dp[i][j+1][k]+dp[i-1][j][k]*(i-j-1-k))%mod;
                    //什么都不加
                dp[i][j][k]=(dp[i][j][k]+dp[i-1][j][k])%mod;
                //最后1
                    //1加0上
                if(i-j-k-1>=1)
                    dp[i][j+2][k]=(dp[i][j+2][k]+dp[i-1][j][k]*(i-j-k-1))%mod;
                    //1加1上
                if(j>=1)
                    dp[i][j][k+1]=(dp[i][j][k+1]+dp[i-1][j][k]*j)%mod;
                    //都不加
                dp[i][j+1][k]=(dp[i][j+1][k]+dp[i-1][j][k])%mod;
                //最后为2
                dp[i][j][k+1]=(dp[i][j][k+1]+dp[i-1][j][k])%mod;
            }

    }
    ll sum = 0;
    for (int j = 0; j <= n; j++)
        for (int k = 0; k <= n - j; k++)
            sum = (sum + dp[n][j][k]) % mod;
    printf("%I64d\n", sum);
    return 0;
}