1087: [SCOI2005]互不侵犯King
Time Limit: 10 Sec Memory Limit: 162 MBSubmit: 3984 Solved: 2323
[ Submit][ Status][ Discuss]
Description
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上
左下右上右下八个方向上附近的各一个格子,共8个格子。
Input
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
Output
方案数。
Sample Input
3 2
Sample Output
16
HINT
DP[ i ][ j ][ k ];表示到第i行已经放了j个国王,且第i行状态为k的方案数
#include<map>
#include<queue>
#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
ll dp[10][100][520],flag[520][520],cnt[520],n,k,ans,mark[520];
int main(void)
{
ll i,j,tmp,k,h,m;
scanf("%lld%lld",&n,&m);
for(i=0;i<(1<<n);i++)
{
if(i&(i>>1))
continue;
tmp=0;
for(j=i;j>0;j>>=1)
tmp+=(j&1);
cnt[i]=tmp;
mark[i]=1;
}
for(i=0;i<(1<<n);i++)
for(j=0;j<(1<<n);j++)
if(!(i&j) && !(i&(j>>1)) && !(j&(i>>1)))
flag[i][j]=1;
for(i=0;i<(1<<n);i++)
if(mark[i])
dp[1][cnt[i]][i]=1;
for(i=1;i<n;i++)
for(j=0;j<(1<<n);j++)
{
if(mark[j]==0)
continue;
for(k=0;k<(1<<n);k++)
{
if(mark[k]==0)
continue;
if(flag[j][k]==0)
continue;
for(h=cnt[j];h+cnt[k]<=m;h++)
dp[i+1][cnt[k]+h][k]+=dp[i][h][j];
}
}
for(i=0;i<(1<<n);i++)
ans+=dp[n][m][i];
printf("%lld\n",ans);
return 0;
}