状压dp [SCOI2005]互不侵犯（洛谷 P1896）

[SCOI2005]互不侵犯

题目描述

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

注：数据有加强（2018/4/25）

输入格式

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

输出格式

所得的方案数

---

刚开始连样例都没搞明白，然后发现国王是一样的，才推出16种；

设方程为：dp[i][j][s]

i 为第几行，j 为该行的国王摆放方式，s 为 1-i 行国王的总数；

状压体现在 j ，也就是国王的摆放方式；

所以状态转移方程为：dp[i][j][s]=sum{dp[i-1][k][s-solve(j)]};

这里的 k 表示上一行国王的摆放方式，solve()这个函数获取的是该行国王的数量；

可想而知，要用到4层循环，细节比较多；

代码：

#include<bits/stdc++.h>
#define ll long long
#define pa pair<int,int>
#define lson k<<1
#define rson k<<1|1
#define inf 0x3f3f3f3f
//ios::sync_with_stdio(false);
using namespace std;
const int N=200100;
const int M=1000100;
const ll mod=998244353;
int n,m;
ll dp[10][2000][100];
int solve(int p){//1的个数 
	int ans=0;
	while(p){
		if(p%2) ans++;
		p/=2;
	}
	return ans;
}
bool judge(int p){
	vector<int>ve;
	int cnt=0;
	while(p){
		cnt++;
		if(p%2) ve.push_back(cnt);
		p/=2;
	}
	for(int i=0;i<(int)ve.size()-1;i++){
		if(ve[i]==ve[i+1]-1) return false;
	}
	return true;
}
int main(){
    ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=0;i<=(1<<n)-1;i++){
		if(judge(i)) dp[1][i][solve(i)]=1;
	}
	for(int i=2;i<=n;i++){
		for(int j=0;j<=(1<<n)-1;j++){//下一行 
			for(int k=0;k<=(1<<n)-1;k++){//上一行 
				if(j&k||(j<<1)&k||(j>>1)&k) continue;
				if(!judge(j)||!judge(k)) continue;
				for(int s=0;s<=m;s++){//国王个数
					if(s<solve(j)||solve(k)>(s-solve(j))) continue;
					dp[i][j][s]+=dp[i-1][k][s-solve(j)];
				}	
			}
		}
	}
	ll ans=0;
	for(int i=0;i<=(1<<n)-1;i++) ans+=dp[n][i][m];
	cout<<ans<<endl;
    return 0;
}