1087: [SCOI2005]互不侵犯King
Time Limit: 10 Sec Memory Limit: 162 MB
Description
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上
左下右上右下八个方向上附近的各一个格子,共8个格子。
Input
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
Output
方案数。
Sample Input
3 2
Sample Output
16
思路:
对于每一行的状态,我们用1表示国王,0表示不放国王。
先枚举每行所有可能的状态,记录下合法状态。然后判断两个状态之间能否转移。若是可以,就更新dp。最后将最后一行的所有状态的方案数相加得ans。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define N 10
#define M 1200
using namespace std;
int n, m, lim;
LL f[N][M][N*N];//f[i][j][k]表示第i行状态为j时,有k个King时的方案数。
int str[M], num[M], cnt;
bool check(int S, int T){
return ((S&T) | (S&(T<<1)) | (S&(T>>1)));
//对于S中每个1(King)其下方三个位置只要有1(King)则不合法
}
int main(){
f[0][1][0] = 1;
scanf("%d%d", &n, &m);
lim = (1<<n)-1;//限度
for(int i=0; i<=lim; i++){
if(( i&(i<<1) ) == 0){//没有两个国王相邻
str[++cnt] = i;//储存状态
int rstate = i, cc = 0;
while( rstate ){
cc += (rstate & 1);
rstate >>= 1;
}
num[cnt] = cc;//当前状态行上的King的个数
}
}
for(int i=1; i<=n; i++){
for(int j=1; j<=cnt; j++){
for(int k=1; k<=cnt; k++)
if( !check(str[j], str[k]) ){//判断j,k状态间是否能够转移
for(int l=num[k]; l<=m; l++){//不能少于当前行个数
f[i][k][l] += f[i-1][j][l-num[k]];
}
}
}
}
LL ans = 0;
for(int i=1; i<=cnt; i++)
ans += f[n][i][m];
cout<<ans<<endl;
return 0;
}