BZOJ 1087: [SCOI2005]互不侵犯King

Description

在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子。

Input

只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N）

Output

方案数。

Sample Input

3 2

Sample Output

16

 

题解

状态dp。状态总数很多，但可行状态很少。这是状压dp中一种常见的问题。记录可行状态是一种常用的优化技巧。

#include<cstdio> 
#include<iostream> 
#include<cstring> 
#include<cstdlib> 
#include<cmath> 
#include<algorithm> 
using namespace std; 
int n,m,S,zz,s[1025],v[1025]; 
int map[1025][1025]; 
long long f[11][1100][90],ans=0; 
int count(int x) 
{ 
    int ct=0; 
    while(x) 
       {if(x&1) ct++; 
        x=x>>1; 
       } 
    return ct; 
} 
bool check(int i,int j) 
{ 
    int x=s[i],y=s[j]; 
    if(x&y) return false; 
    if(x&(y>>1)) return false; 
    if(x&(y<<1)) return false; 
    return true; 
} 
bool judge(int x) 
{ 
    if(x&(x<<1)) return false; 
    if(x&(x>>1)) return false; 
    return true; 
} 
void pre() 
{ 
    S=(1<<n)-1; 
    int i,j; 
    for(i=0;i<=S;i++) 
       {if(judge(i)) 
         {s[++zz]=i; v[zz]=count(i);} 
       } 
    for(i=1;i<=zz;i++) 
    for(j=1;j<=zz;j++) 
       {if(check(i,j)) map[i][j]=map[j][i]=1;} 
} 
void dp() 
{ 
    int i,j,k,l; 
    for(i=1;i<=zz;i++) f[1][i][v[i]]=1; 
    for(i=1;i<n;i++) 
    for(j=1;j<=zz;j++) 
    for(k=0;k<=m;k++) 
       {if(f[i][j][k]) 
           {for(l=1;l<=zz;l++) 
               {if(map[j][l]&&k+v[l]<=m) 
                  f[i+1][l][k+v[l]]+=f[i][j][k]; 
               } 
           } 
       } 
    for(i=1;i<=zz;i++) 
       ans+=f[n][i][m]; 
    printf("%lld\n",ans); 
} 
int main() 
{ 
    scanf("%d%d",&n,&m); 
    pre(); dp(); 
    return 0; 
}