NUIST LevOJ P1826 荷取的基站布局 (状态压缩dp

题目描述

草根妖怪网络科技有限公司正委托河童重工架设若干个基站。

架设地点可以抽象成一个 n * n 的方格阵。每个基站提供信号的范围是其临近的上下左右四格（如果存在）

为了避免资源浪费，需要保证对于任意一个基站，其所在方格不存在其他基站的信号。通俗点讲，就是不能将基站布置在已有其他基站信号覆盖的地方。满足该要求的基站布局方案被成为合法的，反之为不合法的。

下面是两个例子：

 

左图的布局是合法的，而右图的两个基站的信号互相覆盖到了另一个基站，因此是不合法的 (注意：什么都不放也是一个合法的布局方式)

现在，河童重工的工程师荷城荷取要求你帮她算出，在这样一个 n * n 的方格阵中，有多少种不同的合法的布局方案。

输入格式

一个正整数 n，表示方格阵的边长。

输出格式

一个正整数 K，表示在 n * n 的方格阵中，有 K 种不同的合法的布局方案。由于方案数可能很大，请将输出的答案对 1000000007 取模。

输入样例

4

输出样例

1234

数据规模

对于一部分数据，保证 n <= 4

对于所有数据，保证 n <= 10

本题采用状态压缩动态规划算法，用二进制数表示状态，并转换为对应的十进制数存储状态。用二进制数位运算筛选出合法状态，并用位运算判断状态转移的条件，做到行内状态合法，行间状态兼容。

 感兴趣的话可以参考这位up的讲解，和本题只能说一模一样。

https://www.bilibili.com/video/BV1BA411j7Qv/?spm_id_from=444.41.top_right_bar_window_history.content.click&vd_source=3a531974ba11101a65f39bd300c3ece7

AC代码：

#include <iostream>
using namespace std;

typedef long long ll;

const int N = 12;
const int MOD = 1e9 + 7; 
int cnt;	//每行的合法状态数
ll dp[N][1 << N];	//dp(i,j)表示第i行状态为j时，前i行的总方案数 
int state[1 << N];	//每行的合法状态 

int main() {
	int n;
	cin >> n;
	//预处理，得到每行的合法状态 
	for(int i = 0; i < 1<<n; i++) {
		if(!(i & i>>1)) {
			state[cnt++] = i;	//基站左右为空，即为合法 
		}
	}
	dp[0][0] = 1;	//一个基站不放也是一种方案
	for(int i = 1; i <= n + 1; i++) {	//从上到下，枚举每一行，处理当前行时，只需考虑上一行的状态 
		for(int a = 0; a < cnt; a++) {	//state[a]为当前行要处理的状态 
			for(int b = 0; b < cnt; b++) {	//枚举上一行的合法状态state[b]，并判断是否能将状态a转移到状态b 
				if(!(state[b] & state[a])) {	
					//当a状态的基站其正上方位置没有基站时（即state[b]与state[a]做位与运算结果为0），b状态合理 
					dp[i][a] = (dp[i][a] + dp[i - 1][b]) % MOD;	//计算时当前状态a累加上一行所有兼容的状态b 
				}
			} 
		} 
	}
	cout << dp[n + 1][0];	
	return 0;
}