[蓝桥杯 2021 省 AB2] 国际象棋

[蓝桥杯 2021 省 AB2] 国际象棋

题目描述

众所周知， “八皇后” 问题是求解在国际象棋棋盘上摆放 $8$ 个皇后，使得两两之间互不攻击的方案数。已经学习了很多算法的小蓝觉得 “八皇后” 问题太简单了，意犹末尽。作为一个国际象棋迷，他想研究在 $N\times 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

1 2 1

样例输出 #1

2

样例 #2

样例输入 #2

4 4 3

样例输出 #2

276

样例 #3

样例输入 #3

3 20 12

样例输出 #3

914051446

提示

对于 $5\%$ 的评测用例, $K=1$;

对于另外 $10\%$ 的评测用例, $K=2$;

对于另外 $10\%$ 的评测用例, $N=1$;

对于另外 $20\%$ 的评测用例, $N,M\le 6,K\le 5$;

对于另外 $25\%$ 的评测用例, $N\le 3,M\le 20，K\le 12$;

对于所有评测用例, $1\le N\le 6,1\le M\le 100,1\le K\le 20$。

蓝桥杯 2021 第二轮省赛 A 组 I 题（B 组 J 题）。

思路

本道题第一眼看有点类似与小国王和炮兵阵地的题目，再看数据范围，都比较小，ok，我们就用状态压缩dp；

• 首先，我们分析状态表示，我们可以用f[i][a][b][j]表示第i行，第i行的状态a，第i-1行状态b，放了j匹马，然后我们可以画出以下图：
  
• 其次，状态转移方程就很好写了
• $\Rightarrow$ 相邻两列的左右一位的位置上面不能同时有1
  
  $\Rightarrow$ 间隔一列的左右两位的位置上面不能同时有1
  
  a:i - 2列， b:i - 1列， c:i列
  
  (a, b), (b, c)是相邻的一列， (a, c)中间间隔一列
  t： 表示c中的二进制1的个数
  然后就可以进行状态转移了，状态转移方程为：
  $$f[i,b,c,k]+=f[i-1,a,b,k-t]$$

代码

#include<iostream>
#include<algorithm>

using namespace std;

//这道题跟炮兵阵地有点点区别，但区别不是太明显。
//因为炮兵阵地是得看前i-3行，而本道题只需看i-2行即可，因此状态就只看两层即可

const int N = 110,M = 1<<6,K = 22,mod=1e9+7;

int f[N][M][M][K];
int n,m,k;

int get(int x){
    //算某个状态1的个数
    int res=0;
    while(x){
        res++;
        x-=x&-x;
    }
    return res;
}

int main(){
    cin>>n>>m>>k;//n行数、m列数
    f[0][0][0][0]=1;
    //是一样的 一个棋盘N行M列 你转90°就是M行N列了 因为枚举列的状态就要超时了 N是6很小 所以枚举行
    for(int i=1;i<=m;i++){
        for(int a=0;a<(1<<n);a++){
            for(int b=0;b<(1<<n);b++){
                if(b&(a<<2)||a&(b<<2))continue;
                for(int c=0;c<(1<<n);c++){
                    if(c&(b<<1)||b&(c<<1))continue;
                    if(c&(a<<2)||a&(c<<2))continue;
                    int t=get(b);
                    for(int j=t;j<=k;j++){
                        //其实后面加的那个就是能更新a,b状态的状态
                        f[i][a][b][j]=(f[i][a][b][j]+f[i-1][c][a][j-t])%mod;
                    }
                }
            }
        }
    }
    
    int res=0;
    
    for(int a=0;a<(1<<n);a++){
        for(int b=0;b<(1<<n);b++){
            res=(res+f[m][a][b][k])%mod;
        }
    }
    cout<<res;
    
    return 0;
}