对状压dp的一点理解

最近学习了状压dp，既然是dp和其他的dp思想是一致的都是找到转移方程转移方法然后转移就行。不同的在于状压dp利用了二进制把一个状态记录成一个二进制数。由于过去dp练习的多，刷一两道题直接算是入门了。

下面上两道洛谷的入门状压dp：

P1879 [USACO06NOV]玉米田Corn Fields

题面链接：https://www.luogu.org/problem/P1879

题意：给定一个01矩阵，要求找出内部包含的满足没有相邻1的方案数为多少。

思路：n和m的大小为12.故每一行的二进制状态大小就是2的12次方 也就是4000左右。每一行的每一个状态我们都可以用上一行满足条件的状态转移过来。最多12行。故这题可以用复杂度为2的24次方乘12的复杂度算出来。内部有大量条件不合法稍微减点枝，这个大小是绰绰有余的。具体细节我的代码有注释。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
int n,m;
ll dp[15][(1<<15)];
int a[15][15];
int f[(1<<15)];

int mod = (int)1e8;

int check(int r,int num){
    for(int i=1;i<=m;i++){
        if(num%2 && !a[r][i])   return 0;
        num >>= 1;
    }
    return 1;
}
int main()
{
    cin >> n >> m;

    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            scanf("%d",&a[i][j]);
        }
    }

    int maxn = 1<<m;
    /*
        maxn是所有每一行最多有多少状态
        例如m=3
        000 001 010 011 100 101 110 111

        我们先预处理每一行合法和不合法的排列
    */
    for(int i=0;i<maxn;i++){
        if(i&(i<<1))   f[i] = 1;
        // 有相邻的1,不可行
    }

    dp[0][0] = 1;
    for(int i=1;i<=n;i++){
        for(int j=0;j<maxn;j++){
            if(f[j])    continue;
            if(!check(i,j)) continue;

            for(int k=0;k<maxn;k++){
                if(k & j)   continue;
                dp[i][j] += dp[i-1][k];
                dp[i][j] %= mod;
            }
        }
    }
    int ans = 0;
    for(int i=0;i<maxn;i++) ans = (ans + dp[n][i])%mod;

    cout << ans;

    return 0;
}

P1896 [SCOI2005]互不侵犯

题面链接：https://www.luogu.org/problem/P1896

题意：给定一个n*n的棋盘。在上面放k个王。满足这k个王无法相互攻击。王可以攻击他相邻的8个格子。

思路：相当于上一题的一个加强版。方法一样。多了一维度保存已经使用的王个数即可。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int n,K;
int f[(1<<10)];
ll dp[10][(1<<10)][100];

int check(int a,int b){
    if(a&b) return 0;
    if((a<<1)&b) return 0;
    if((a>>1)&b) return 0;
    return 1;
}

int main()
{
    cin >> n >> K;

    int maxn = 1<<n;
    for(int i=0;i<maxn;i++){
        if((i<<1)&i)    f[i] = 1;
    }


    dp[0][0][0] = 1;
    for(int i=1;i<=n;i++){
        for(int j=0;j<maxn;j++){
            if(f[j])    continue;
            for(int k=0;k<maxn;k++){
                if(!check(k,j)) continue;

                int num1 = 0;
                int tmp = j;

                while(tmp){
                    num1 += tmp%2;
                    tmp >>= 1;
                }

                for(int cnt = 0;cnt <= K;cnt++){
                    if(cnt-num1 < 0) continue;
                    dp[i][j][cnt] += dp[i-1][k][cnt-num1];
                }

            }
        }
    }

    ll ans = 0;
    for(int i=0;i<maxn;i++){
        ans += dp[n][i][K];
    }
    cout << ans;



    return 0;
}

多多接触一下新算法有好处的。算法使用的思想对自己思考别类型的题面也是一种提升。以前忽视了这个重要性真是血马亏。