状压dp第一弹

互不侵犯King HYSBZ - 1087

很基础的状鸭dp

用二进制枚举出每一行可能出现的总状态：对于同一行内的某一状态x，!(x) & (x << 1)即可以保证King们在一行内不会攻击。而对于每一行的某一状态x和他的下一行的某一状态y， 需要满足!(x & y), !(x & (y << 1)), !((x << 1)& y), 即king分别在上下，斜向方向上保持互不攻击.

#include <bits/stdc++.h>

using namespace std;

long long int state[1 << 10];
long long int all_state;
long long int dp[10][1 << 10][82];//dp[i][j][k] 表示第i行的第j个状态下场上存在k个king的方案数
int cnt[1 << 10];//记录每种状态下此行内king的个数

void getState(int n)
{
	int N = 1 << n;
	for (int i = 0; i < N; ++i)
	{
		if (!(i & (i << 1)))//同一行内不会攻击的状态
		{
			state[all_state] = i;
			for (int j = 0; j < n; ++j)
				if ((1 << j) & i)
					++cnt[all_state];
			++all_state;
		}
	}
}

bool fit(int x, int y)
{
	if (x & y)
		return 0;
	if (x & (y << 1))
		return 0;
	if ((x << 1) & y)
		return 0;
	return 1;
}

int main()
{
	int n, k;
	cin >> n >> k;
	getState(n);
	for (int i = 0; i < all_state; ++i)
		dp[0][i][cnt[i]] = 1;
	long long int ans = 0;
	for (int i = 1; i < n; ++i)
	{
		for (int j = 0; j < all_state; ++j)
		{
			for (int l = 0; l < all_state; ++l)
			{
				if (fit(state[j], state[l]))
				{
					for (int m = 1; m <= k; ++m)//其实我觉得m应该从零开始。。。。TWT
					{
						if (m + cnt[j] > k)
							break;
						dp[i][j][m + cnt[j]] += dp[i - 1][l][m];
					}
				}
			}
		}
	}
	for (int i = 0; i < n; ++i)
	{
		for (int j = 0; j < all_state; ++j)
		{
			ans += dp[i][j][k];
		}
	}
	cout << ans << endl;
	return 0;
}