题目描述 在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
输入输出格式 输入格式: 只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
输出格式: 所得的方案数
输入输出样例:
3 2
output
16
思路: N的个数是十分小的 (即使这样爆搜也会T飞!!) 所以用一波 状压DP!
设f[i][j][k]代表前i行放了k个,排列方式是j,(j的二进制形式就是排列方式)。
可以 (必须) 先预处理出一行的可行排列情况: 1 2 3 4 5 6 7 8 9 10 11 void dfs (int w,int sum,int num) { if (num>=n) { sit[++cnt]=w; gs[cnt]=sum; return ; } dfs (w,sum,num+1 ); dfs (w+(1 <<num),sum+1 ,num+2 ); }
那么根据题意,这个方案与另一个方案是否冲突 就显而易见啦 sit[j]&sit[k]==1 上下相邻 (sit[j]<<1)&sit[k]==1 斜着相邻 sit[j]&(sit[k]<<1) ==1 同上
DP+判断 code: 第一行:到第几行了 第二行:这一行的的排列情况 第三行:这一行的的排列情况可以从上一行的哪几个状态转移过来。
1 2 3 4 5 6 7 8 9 for (int i=2 ;i<=n;i++) for (int j=1 ;j<=cnt;j++) for (int k=1 ;k<=cnt;k++) { if (sit[j]&sit[k])continue ; if ((sit[j]<<1 )&sit[k]) continue ; if (sit[j]&(sit[k]<<1 )) continue ; for (int s=K;s>=gs[j];s--) f[i][j][s]+=f[i-1 ][k][s-gs[j]]; }
完整代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 #include <bits/stdc++.h> using namespace std;#define maxn 2005 int cnt,n,K;int sit[maxn],gs[maxn];long long f[10 ][maxn][105 ];long long ans;void dfs (int w,int sum,int num) { if (num>=n) { sit[++cnt]=w; gs[cnt]=sum; return ; } dfs (w,sum,num+1 ); dfs (w+(1 <<num),sum+1 ,num+2 ); } int main () { scanf ("%d%d" ,&n,&K); dfs (0 ,0 ,0 ); for (int i=1 ;i<=cnt;i++) f[1 ][i][gs[i]]=1 ; for (int i=2 ;i<=n;i++) for (int j=1 ;j<=cnt;j++) for (int k=1 ;k<=cnt;k++) { if (sit[j]&sit[k])continue ; if ((sit[j]<<1 )&sit[k]) continue ; if (sit[j]&(sit[k]<<1 )) continue ; for (int s=K;s>=gs[j];s--) f[i][j][s]+=f[i-1 ][k][s-gs[j]]; } ans=0 ; for (int i=1 ;i<=cnt;i++) ans+=f[n][i][K]; printf ("%lld" ,ans); return 0 ; }
注意开long long。。。