1. 程式人生 > >狀態壓縮dp學習 + poj3254(最簡單的狀態壓縮dp)

狀態壓縮dp學習 + poj3254(最簡單的狀態壓縮dp)

初學狀壓就找個比較簡單的dp,應該是最好想的狀態壓縮了。

狀態壓縮dp也就是如果正常開陣列,必然會導致記憶體過大,所有需要另外找一種辦法來代替這個問題,那麼就用二進位制來替代,這應該就是對於狀態壓縮最通俗的解釋了。關於一些二進位制的操作比如

1 如何判斷數字x第i位是否為1    1<<(i-1) & x

2 將一個數字x二進位制下第i位更改成1  x= x|(1<<(i-1))

3 把一個數字二進位制下最靠右的第一個1去掉  x=x&(x-1)  (BIT裡面我記得用到的就是這個)

總之,二進位制真的是一個十分巧妙的東西,也是一個非常好用的東西

然後是經典的TSP(旅行商問題)

程式碼什麼的白書上都有,然後白書上一些對於位運算的操作

poj3254

題目大意是:給一個n*m的矩陣,上面1表示肥沃的土地,0表示貧瘠的土地,然後要讓牛必須站在肥沃的土地上,又不能相鄰(可以整個都不站牛)

題解:那就用二進位制來表示整個土地的樣子,狀態轉移是dp[i][j] = sum(dp[i-1][0-cnt]) 表示第i行的站法是由i-1行所有可能的站法並且兩行不能衝突的和

有一個技巧就是如果判斷一個數字是否有相鄰的兩個1  x&(x<<1) 如果是0則表示沒有1則表示有。

#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#include <set>
#include <stack>
#include <cmath>
#define fuck() (cout << "--------------------------------" << endl)
#define mod 100000000
using namespace std;
const int maxn = 100000 + 5;
const int inf = 0x3f3f3f3f;
int ok[maxn];
int maps[maxn];
int dp[13][maxn];
bool judge(int i)
{
    return i & (i << 1);
}
bool judge2(int x, int y)
{
    return maps[x] == (maps[x]|ok[y]);//0 表示不可以 1 表示行
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF && n)
    {
        memset(maps,0,sizeof(maps));
        for(int i=0; i<n; i++)
            for(int j=0; j<m; j++)
            {
                int x;
                scanf("%d",&x);
                if(x == 1)
                    maps[i] += (1 << j);//用二進位制儲存每一行肥沃土地的樣子 100101 轉化成十進位制
            }
        int cnt = 0;
        for(int i=0; i<(1<<m); i++)//找出所有滿足條件的數字
            if(!judge(i))
                ok[cnt++] = i;
        memset(dp,0,sizeof(dp));
        for(int i=0; i<cnt; i++)
        {
            if(judge2(0,i))
                dp[0][i] = 1; //確定第一行有哪幾種可以放的方式
        }
        for(int i=1; i<n; i++)
        {
            for(int j=0; j<cnt; j++)
            {
                if(!judge2(i,j)) continue;
                for(int k=0; k<cnt; k++)//表示這種擺的方式在第i行可以,還要和第i-1行判斷
                {
                    if(!(ok[j]&ok[k]))
                        dp[i][j] += dp[i-1][k];
                }
            }
        }
        long long  ans = 0;
        for(int i=0; i<cnt; i++)
        {
            ans += dp[n-1][i];
            ans %= mod;
        }
        printf("%lld\n",ans);
    }
}