DP - 状压DP - SGU - 223 - Little Kings

DP - 状压DP - SGU - 223 - Little Kings

在 n×n 的棋盘上放 k 个国王，国王可攻击相邻的 8 个格子，求使它们无法互相攻击的方案总数。

输入格式
共一行，包含两个整数 n 和 k。

输出格式
共一行，表示方案总数，若不能够放置则输出0。

数据范围
1≤n≤10,
0≤k≤n²

输入样例：
3 2
输出样例：
16

---

分析：

$本题与——$《蒙德里安的梦想》$相似，只是摆放的物品占据的范围不同。$

$经过分析，容易发现：由于每个国王仅占据相邻的八个方格，故每一行摆放的方案仅与上一行的状态有关。$

$$\begin{gathered} 那么我们可以通过一个二进制数s来表示第i行的状态，1表示该位置放一个国王，0表示不放。 \\ 由于国王的数量有限制，因此我们的状态表示需要三维。 \end{gathered}$$

$状态表示，f[i][j][s]：已经摆放了前i行，且用了j个物品，且第i行的状态是s的总方案数。$

$$\begin{gathered} 状态计算： \\ 假设第i行的状态是a，第i-1行的状态是b，若a能够由b转移而来，需要满足： \\ ①、第i-1行本身是合法的：二进制数b不能存在连续的1。 \end{gathered}$$

$$\begin{gathered} ②、状态a和状态b不能存在冲突： \\ Ⅰ、第i行放国王的位置的正上方不能有国王，需满足a\&b=0。 \end{gathered}$$

$$\begin{gathered} Ⅱ、第i行放国王的位置的斜上方不能有国王，需满足a|b表示的二进制数不能有连续的1。 \\ 这一条件同时也避免了第i行出现连续的1。 \end{gathered}$$

$$\begin{gathered} 若满足以上两个条件，转移方程：f[i][j][a]+=f[i-1][j-count(a)][b]， \\ 其中count(a)表示状态a中1的个数，即第i行摆放国王的数量。 \end{gathered}$$

具体落实：

$①、先预处理每一行的所有合法状态(不能含有连续的1)，同时记录下这些状态当中1的个数(状态转移的时候要用)。$

$②、将所有互不影响的状态预处理出来，用head数组做一个映射。$

$③、计算每个状态的方案总数。$

注意：

$①、初始化边界。对每一层而言，摆放的国王数量为0时也是一种合法方案，因此国王数量j从0开始循环。$

$$\begin{gathered} ②、最后要输出第n层的方案总数之和，即\sum f[n][m][s]， \\ 我们可以在计算时算到n+1层，最后输出f[n+1][m][0]，这是等价的。 \\ 因为状态为0可由前一层的所有合法状态转移而来，相当于在计算方案数的过程中就做了一遍求和。 \end{gathered}$$
代码：

#include<iostream>
#include<algorithm>
#include<vector>

#define ll long long

using namespace std;

const int N=12,M=1<<10,K=110;

int n,m;
ll f[N][K][M];
int cnt[M];
vector<int> state;
vector<int> head[M];

bool check(int x)
{
    for(int i=0;i<n-1;i++)
        if((x>>i)&1 && (x>>i+1)&1)
            return false;
    return true;
}

int Count(int x)
{
    int res=0;
    for(int i=0;i<n;i++) res+=(x>>i)&1;
    return res;
}

int main()
{
    cin>>n>>m;
    
    for(int i=0;i<1<<n;i++)
        if(check(i))
        {
            state.push_back(i);
            cnt[i]=Count(i);
        }
    
    int len=state.size();
    for(int i=0;i<len;i++)
        for(int j=0;j<len;j++)
        {
            int a=state[i],b=state[j];
            if((a&b)==0 && check(a|b))
                head[i].push_back(j);
        }
    
    f[0][0][0]=1;
    for(int i=1;i<=n+1;i++)
        for(int j=0;j<=m;j++)
            for(int a=0;a<len;a++)
                for(int b:head[a])
                {
                    int c=cnt[state[a]];
                    if(j>=c) f[i][j][state[a]]+=f[i-1][j-c][state[b]];
                }
            
    cout<<f[n+1][m][0]<<endl;
    
    return 0;
}

---