bzoj1087

又一道状压dp

查了半天错 又被位运算的优先级坑了

#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define b(a) (1<<(a))
#define clr(a,x) memset(a,x,sizeof(a))
#define rep(i,l,r) for(int i=l;i<r;i++)
typedef long long ll;
using namespace std;
int read()
{
    char c=getchar();
    int ans=0,f=1;
    while(!isdigit(c)){
        if(c=='-') f=-1;
        c=getchar();
    }
    while(isdigit(c)){
        ans=ans*10+c-'0';
        c=getchar();
    }
    return ans*f;
}
const int maxn=10,maxk=100;
int n,k;
ll d[maxn][maxk][b(maxn)];
bool legal(int s)
{
    while(s){
        if(s&1){
            if(s&b(1)) return 0;
        }
        s=s>>1;
    }
    return 1;
}
int cnt(int a)
{
    int sum=0;
    while(a){
        if(a&1) sum++;
        a=a>>1;
    }
    return sum;
}
ll dp(int a,int b,int s)
{    
    if(d[a][b][s]!=-1) return d[a][b][s];
    if(b==0&&cnt(s==0)) return d[a][b][s]=1;
    ll sum=0;
    if(b-cnt(s)>=0){
        rep(i,0,b(n)){
            if(legal(i)&&legal(i|s)&&((i&s)==0))
                sum+=dp(a-1,b-cnt(s),i);
        }
    }    
    return d[a][b][s]=sum;
}
int main()
{
    n=read(),k=read();
    clr(d,-1);
    rep(i,0,b(n))
        rep(j,0,k+1)
            d[1][j][i]=0;
    rep(i,0,b(n)){
        if(legal(i)) d[1][cnt(i)][i]=1;
    } 
    ll ans=0;
    rep(i,0,b(n)) 
    if(legal(i))
    ans+=dp(n,k,i);
    printf("%lld
",ans);
    return 0;
}

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1977  Solved: 1174
[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

 

Source

 

[Submit][Status][Discuss]