码迷,mamicode.com
首页 > 其他好文 > 详细

【状压dp】【bzoj 1087】【SCOI 2005】互不侵犯King

时间:2015-07-29 19:29:29      阅读:141      评论:0      收藏:0      [点我收藏+]

标签:

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1991  Solved: 1185

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

题解:
状压dp,设f[i][j][k]表示前i行一共放了j个,第i行的状态是k(k是对这一行压出来的二进制数)。
转移:

f[i][j][k]=Σalli=0f[i?1][j?count(k)][s]

其中all是所有状态个数,count(k)是计算k中1的个数,也就是放了几个,s是i-1行的状态。
转移时要保证s状态合法且不与k状态冲突,可以用到位运算来快速判断。

Code:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int N,K,all;
long long ans,f[20][100][520];
int count(int x){
    int s=0;
    for (int i=x; i>0; i>>=1)
        if (i&1) s++;
    return s;
}
void dp(){
    all=(1<<N)-1; ans=0;
    memset(f,0,sizeof(f));
    for (int i=0; i<=all; i++)
        if (!(i&(i<<1)) && !(i&(i>>1)))
            f[1][count(i)][i]=1;
    for (int i=2; i<=N; i++)
        for (int j=0; j<=K; j++)
            for (int k=0; k<=all; k++)
                if (!(k&(k<<1)) && !(k&(k>>1))){
                    int tot=count(k);
                    if (j<tot) continue;
                    for (int s=0; s<=all; s++)
                        if (!(s&(s<<1)) && !(s&(s>>1)) && !(s&k) && !(s&(k<<1)) && !(s&(k>>1)))
                            f[i][j][k]+=f[i-1][j-tot][s];
                }
    for (int i=0; i<=all; i++)
        if (!(i&(i<<1)) && !(i&(i>>1)))
            ans+=f[N][K][i];
}
int main(){
    scanf("%d%d",&N,&K);
    dp(); printf("%lld\n",ans);
    return 0;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

【状压dp】【bzoj 1087】【SCOI 2005】互不侵犯King

标签:

原文地址:http://blog.csdn.net/morestep/article/details/47130581

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!