第十二届蓝桥杯 ——国际象棋

题目描述
众所周知，“八皇后” 问题是求解在国际象棋棋盘上摆放 8个皇后，使得两两之间互不攻击的方案数。

已经学习了很多算法的小蓝觉得 “八皇后” 问题太简单了，意犹未尽。

作为一个国际象棋迷，他想研究在 N × M的棋盘上，摆放 K 个马，使得两两之间互不攻击有多少种摆放方案。

由于方案数可能很大，只需计算答案除以 1000000007 (即10^9+7) 的余数。

如下图所示，国际象棋中的马摆放在棋盘的方格内，走 “日” 字，位于 ( x , y ) 格的马（第 x 行第 y 列）可以攻击

( x + 1 , y + 2 ) 、 ( x + 1 , y − 2 ) 、 ( x − 1 , y + 2 ) 、 ( x − 1 , y − 2 ) 、 ( x + 2 , y + 1 ) 、 ( x + 2 , y − 1 ) 、 ( x − 2 , y + 1 ) 和 ( x − 2 , y − 1 )

共 8 个格子。

输入格式
输入一行包含三个正整数 N , M , K分别表示棋盘的行数、列数和马的个数。

输出格式
输出一个整数，表示摆放的方案数除以 1000000007(即10^9+7) 的余数。

输入样例1
1 2 1

输出样例1
2

输入样例2
4 4 3

输出样例2
276

输入样例3
3 20 12

输出样例3
914051446

数据范围
对于 5 55% 的评测用例，K = 1 K=1K=1；
对于另外 10 1010% 的评测用例，K = 2 K=2K=2；
对于另外 10 1010% 的评测用例，N = 1 N=1N=1；
对于另外 20 2020% 的评测用例，N , M ≤ 6 ， K ≤ 5 N,M≤6，K≤5N,M≤6，K≤5；
对于另外 25 2525% 的评测用例，N ≤ 3 ， M ≤ 20 ， K ≤ 12 N≤3，M≤20，K≤12N≤3，M≤20，K≤12；
对于所有评测用例，1 ≤ N ≤ 6 ， 1 ≤ M ≤ 100 ， 1 ≤ K ≤ 20 1≤N≤6，1≤M≤100，1≤K≤201≤N≤6，1≤M≤100，1≤K≤20。
 

思路：

状态压缩DP，dp[i][j][k][t]表示第i行共有j个马并且i-1行的状态为k，i-行的状态为t时摆放方式。

#include <bits/stdc++.h>
using namespace std;

const int M = 1 << 6, MOD = 1e9 + 7;

int n, m, K;
int dp[110][21][M][M];

int count(int x)
{
	int res = 0;
	while (x)
	{
		res++;
		x &= (x-1);
	}
	return res;
}

int main()
{
    cin >> n >> m >> K;
    
    dp[0][0][0][0] = 1;
    for (int i = 1; i <= m; i ++)
	{
        for (int k = 0; k <= K; k ++)
		{
            for (int b = 0; b < 1 << n; b ++)
			{
                for (int a = 0; a < 1 << n; a ++)
                {
                    if (b & (a << 2) || b & (a >> 2))
					{
						continue;
					}
                    for (int c = 0; c < 1 << n; c ++)
                    {
                        if (c & (a << 1) || c & (a >> 1)) 
						{
							continue;
						}
                        if (c & (b << 2) || c & (b >> 2)) 
						{
							continue;
						}
						int temp = count(a);
                        if (k >= temp)
						{
                            dp[i][k][b][a] = (dp[i][k][b][a] + dp[i - 1][k - temp][c][b]) % MOD;
						}
                    }
                }
			}
		}
	}
                
    int ans = 0;
    for (int i = 0; i < 1 << n; i ++)
	{
        for (int j = 0; j < 1 << n; j ++)
		{
            ans = (ans + dp[m][K][i][j]) % MOD;
		}
	}
            
    cout << ans << endl;
    return 0;
}