DP(状压专题八)

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

>> face <<

Strategy:状压DP, i是一个表示二进制的十进制数,代表当前行的状态

状态: dp[i][r][k] - > r行状态为i且用了k个棋子的有效摆放方案

目标:

$$\sum _{i=0}^{i<(1<<n)-1}\sum _{r=1}^{r\le n}dp[i][r][k]$$

边界: 主要针对第一行

$$dp[i][1][onecount(i)]=1,i\in 合法状态$$

合法判断: 利用矩阵判断当前行是否合法，上一行和当前行状态的&等于0 左移右移再判断

转移方程:

$$dp[i][r][k+cnt(i)]+=dp[j][r-1][k]$$

attention: 状态转移的时候一定操你大爷的敢不注意i和j不要再弄混了

双倍经验:

• 调试经验 :
• #define bin(x) cout << #x << " is " << bitset<sizeof(int)*2>(x) << endl 输出二进制数;
• 带有限制条件的状态最好设计成已经使用了多少空间而不是还剩多少空间
• 状态转移的时候一定他要妈的 注意i和j不要再弄混了

#include <bits/stdc++.h>
#include <bits/extc++.h>
#define oo 0x3f3f3f3f
#define ll long long
#define db double
#define all(a) a.begin(), a.end()
#define met(a, b) memset(a, b, sizeof(a))
#define what_is(x) cerr << #x << " is " << x << endl
#define _rep(i, a, b) for (int i = (a); i <= (b); ++i)
#define _rev(i, a, b) for (int i = (a); i >= (b); --i)
#define _for(i, a, b) for (int i = (a); i < (b); ++i)
#define lowbit(x) x &(-x)
#define bin(x) cout << #x << " is " << bitset<sizeof(int)*2>(x) << endl
#define pi acos(-1.0)
using namespace std;
using namespace __gnu_pbds;
const int maxn = 20;

const db eps = 1e-8;
int n, k;
ll dp[1 << 10][10][101];
bool valid[1 << 10];


int main()
{
    cin >> n >> k;
    
    _rep(i, 0, (1 << n) - 1){
        bool  is_valid = 1;
        _for(j, 1, n){
            if(i >> j - 1 & 1 && i >> j & 1)
                is_valid = 0;
            if(!is_valid)
                break;
        }
        valid[i] = is_valid ? 1 : 0;
        if(is_valid){
            int cnt = __builtin_popcount(i);
            if(cnt <= k)
                dp[i][1][cnt] = 1;
            
        }
    }


    
    _rep(r, 2, n){
        _rep(i, 0, (1 << n) - 1){ //当前行
            if(!valid[i])
                continue;
            _rep(j, 0, (1 << n) - 1){//上一行
                //bin(i), bin(j);
                if(!valid[j] || j & i || j << 1 & i ||  j >> 1 & i)
                    continue;
                int cnt = __builtin_popcount(i);//当前行去掉的国王
                for(int used_k = 1; used_k + cnt <= k; used_k++){
                    
                    dp[i][r][used_k + cnt] += dp[j][r - 1][used_k];
                    //what_is(dp[i][r][used_k + cnt]), what_is(dp[j][r - 1][used_k]), what_is(used_k), what_is(cnt + used_k);
                }				
            }
        }
    }
    ll ans = 0;
    _rep(i, 0, (1 << n) - 1){
        if(!valid[i])continue;
        _rep(r, 1, n){
            ans += dp[i][r][k];	
        }
    }
    cout << ans << endl;
}

第一次回顾:其实完全可以把可行状态全部先筛出来, 这样又简单又快, 代码量可以大幅度降低

都回顾了, 还不记得开long long

#include <bits/stdc++.h>
#include <bits/extc++.h>
#define _rep(i, a, b) for (int i = (a); i <= (b); ++i)
#define _rev(i, a, b) for (int i = (a); i >= (b); --i)
#define _for(i, a, b) for (int i = (a); i < (b); ++i)
#define _rof(i, a, b) for (int i = (a); i > (b); --i)
#define ll long long
#define db double
#define oo 0x3f3f3f3f
#define eps 0.00001
#define all(x) x.begin(), x.end()
#define met(a, b) memset(a, b, sizeof(a))
#define what_is(x) cerr << #x << " is " << x << endl
#define lowbit(x) x &(-x)
using namespace std;
const int maxn = 102;
const int mod = 9999973;
ll dp[10][1 << 10][93], n, k, num[1 << 10];
vector<int> valid;
int main()
{
	ios::sync_with_stdio(0);
	cin >> n >> k;
	_for(i, 0, (1 << n))
	{
		if (i & i >> 1 || i & i << 1)
			continue;
		valid.push_back(i);
		num[i] = __builtin_popcount(i);
	}
	for (auto i : valid)
	{
		dp[1][i][num[i]] = 1;
	}
	_rep(r, 2, n)
	{
		for (auto cur : valid)
		{
			for (auto last : valid)
			{
				if (cur & last || cur >> 1 & last || cur << 1 & last)
					continue;
				for(int used_k = 0; used_k + num[cur] <= k; used_k++)
				{
					dp[r][cur][used_k + num[cur]] += dp[r - 1][last][used_k];
				}
			}
		}
	}
	ll ans = 0;

	for (auto i : valid)
	{
		ans += dp[n][i][k];
	}

	cout << ans << endl;
}

不过还好, 提升不明显, 但是代码量可以少很多 (忽略掉中间内阁没有long long 的wa) 再次提醒 == 不开longlong 见祖宗==