51node1371 填数字(dp)

51node1371

题目

思路

代码

#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
[j][k]) % mod;
printf("%I64d\n", sum);
return 0;
}