BZOJ 1087-互不侵犯King（状压DP）

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 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;}