HYSBZ - 4806_炮_DP

最新推荐文章于 2019-01-23 21:14:59 发布

yuege38

最新推荐文章于 2019-01-23 21:14:59 发布

阅读量213

点赞数

分类专栏： DP 文章标签： dp

本文链接：https://blog.csdn.net/yuege38/article/details/78419629

版权

DP 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

题意

在 N * M 的棋盘上放任意个象棋棋子炮，要求两两之间不能相互攻击。求有多少中方案。

思路

每一行或每一列最多有两个，也就是说每一个列有三种状态 0 1 2，状态相同的列无差别。
dp[i][j][k][t] 放完 i 行后 j 列有 0 个， k 列有 1 个， t 列有两个。
j + k + t == M，可以化去一维。
dp[i][j][k] 放完前 i 行后 j 列有 0 个， k 列有 1 个。
转移比较明显。

链接

https://vjudge.net/contest/176068#problem/K

代码

#include<cstdio>
#include<iostream>
#include<cstring>

using namespace std;

typedef long long LL;

const int maxn = 100 + 10;
const int mod  = 999983;

int n, m;
LL dp[maxn][maxn][maxn];

int main()
{
    //freopen("in.txt", "r", stdin);

    while(scanf("%d %d", &n, &m) != EOF)
    {
        memset(dp, 0, sizeof dp);
        dp[0][m][0] = 1;

        for(int i= 1; i<= n; i++)
        {
            for(int j= 0; j<= m; j++)
            {
                for(int k= 0; k<= m - j; k++)
                {
                    dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j][k]) % mod;
                    if(j > 0) dp[i][j - 1][k + 1] = (dp[i][j - 1][k + 1] + dp[i - 1][j][k] * j) % mod;
                    if(k > 0) dp[i][j][k - 1] = (dp[i][j][k - 1] + dp[i - 1][j][k] * k) % mod;
                    if(j > 0 && k > 0) dp[i][j - 1][k] = (dp[i][j - 1][k] + dp[i - 1][j][k] * j * k) % mod;
                    if(j > 1) dp[i][j - 2][k + 2] = (dp[i][j - 2][k + 2] + dp[i - 1][j][k] * j * (j - 1) / 2) % mod;
                    if(k > 1) dp[i][j][k - 2] = (dp[i][j][k - 2] + dp[i - 1][j][k] * k * (k - 1) / 2) % mod;
                }
            }
        }

        int res = 0;
        for(int j= 0; j<= m; j++)
        {
            for(int k= 0; k<= m - j; k++)
            {
                res = (res + dp[n][j][k]) % mod;
            }
        }

        cout << res << endl;
    }

    return 0;
}