[SCOI2005]互不侵犯 解题报告
题目描述
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
输入输出格式
输入格式:
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
输出格式:
所得的方案数
输入输出样例:
input
3 2
output
16
思路:
N的个数是十分小的 (即使这样爆搜也会T飞!!)
所以用一波 状压DP!
设f[i][j][k]代表前i行放了k个,排列方式是j,(j的二进制形式就是排列方式)。
可以 (必须) 先预处理出一行的可行排列情况:
void dfs(int w,int sum,int num)//分别为排列情况,放的点的个数,到底几个点了
{
if(num>=n)
{
sit[++cnt]=w;
gs[cnt]=sum;
return;
}
dfs(w,sum,num+1);//不防这个点
dfs(w+(1<<num),sum+1,num+2);//放这个点,那么下一个点不能放
}
那么根据题意,这个方案与另一个方案是否冲突 就显而易见啦
sit[j]&sit[k]==1 上下相邻
(sit[j]<<1)&sit[k]==1 斜着相邻
sit[j]&(sit[k]<<1) ==1 同上
DP+判断 code:
第一行:到第几行了
第二行:这一行的的排列情况
第三行:这一行的的排列情况可以从上一行的哪几个状态转移过来。
for(int i=2;i<=n;i++)
for(int j=1;j<=cnt;j++)
for(int k=1;k<=cnt;k++)
{
if(sit[j]&sit[k])continue;
if((sit[j]<<1)&sit[k]) continue;
if(sit[j]&(sit[k]<<1)) continue;
for(int s=K;s>=gs[j];s--) f[i][j][s]+=f[i-1][k][s-gs[j]];
}
完整代码:
#include<bits/stdc++.h>
using namespace std;
#define maxn 2005
int cnt,n,K;
int sit[maxn],gs[maxn];
long long f[10][maxn][105];
long long ans;
void dfs(int w,int sum,int num)
{
if(num>=n)
{
sit[++cnt]=w;
gs[cnt]=sum;
return;
}
dfs(w,sum,num+1);
dfs(w+(1<<num),sum+1,num+2);
}
int main()
{
scanf("%d%d",&n,&K);
dfs(0,0,0);
for(int i=1;i<=cnt;i++)
f[1][i][gs[i]]=1;
for(int i=2;i<=n;i++)
for(int j=1;j<=cnt;j++)
for(int k=1;k<=cnt;k++)
{
if(sit[j]&sit[k])continue;
if((sit[j]<<1)&sit[k]) continue;
if(sit[j]&(sit[k]<<1)) continue;
for(int s=K;s>=gs[j];s--) f[i][j][s]+=f[i-1][k][s-gs[j]];
}
ans=0;
for(int i=1;i<=cnt;i++) ans+=f[n][i][K];
printf("%lld",ans);
return 0;
}
注意开long long。。。