[SCOI2005]互不侵犯(状压DP)

题意:给定一个N * N的棋盘,放置K个国王,使他们互相攻击不到对方,共有多少种方案。国王可以攻击上下左右,左上左下,右上右下附近的一格,共8格。
数据范围:1<=N<=9, 0<=K<=N * N.

因为N的范围很小,而且每一行对应的一个方案可以用一个二进制数表示,所以容易想到用状压DP。又因题目有限制要用多少个国王,所以我们需要记录一下国王的个数。
于是,我们可以定义DP状态:
dp[i,j,k]表示第i行采用j方案,国王总个数为k时的总方案数。
那么如何推出状态转移方程呢?
状压DP在于枚举,我们先想想怎么枚举。
设置can[i],记录在不考虑行与行之间影响的情况下,每行可以采取的方案。那么,因为每个方案对应有用到一定个数的国王,所以再设置一个num[i],记录采用can[i]方案时用了多少个国王。
初始完can[i],num[i]后,首先初始化第一行:dp[1,can[i],num[i]]=1.
对于i ← 2 ~ N行,假如i-1行采用了can[j]方案,i行采用了can[k]方案,那么在can[k]与can[j]不冲突的情况下,不难推出:dp[i,can[k],num[k]+p]+=dp[i,can[j],p],其中0<=p<=K-num[k],即表示第i行采用can[k]方案时在前一行采用can[j]方案,且国王总个数为p的情况下推出来的。

代码如下:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=1<<9;
ll dp[10][maxn][90];
int can[maxn],num[maxn];

int count(int x)
{
	int sum=0;
	while(x)
	{
		if(x%2)	sum++;
		x/=2;
	}
	return sum;
}

int main()
{
	int N,K;
	cin>>N>>K;
	int cnt=0,tot=1<<N;
	for(int i=0;i<tot;i++)
	{
		if(!(i&(i<<1)))	//同行各个国王不能相邻
		{
			int c=count(i);
			if(c<=K)
			{
				can[++cnt]=i;
				num[cnt]=c;
			}
		}
	}
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=cnt;i++)	
		dp[1][can[i]][num[i]]=1;
	for(int i=2;i<=N;i++)
		for(int j=1;j<=cnt;j++)
		{
			for(int k=1;k<=cnt;k++)
			{
				if(!(can[k]&can[j])&&!((can[k]<<1)&can[j])&&!((can[k]>>1)&can[j]))	//相邻两行不冲突
				{
					for(int p=0;p<=K-num[k];p++)		//枚举
						dp[i][can[k]][num[k]+p]+=dp[i-1][can[j]][p];
				}
			}
		}
	ll sum=0;
	for(int i=1;i<=cnt;i++)	//最终答案落在sum(dp[N,can[1~cnt],K])
		sum+=dp[N][can[i]][K];
	cout<<sum<<endl;
 } 
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值