banner
cos

cos

愿热情永存,愿热爱不灭,愿生活无憾
github
tg_channel
bilibili

狀壓dp——模板&分析&例題(存用)

狀壓講解部分參考博客狀壓 DP 詳解(位運算)

一、什麼是狀壓 dp?#

這裡是引用

狀壓 dp,即將狀態壓縮成 2 進制來保存,如矩陣中將一行的狀態壓縮成一個二進制串,如 dp [S][v] 中,S 可以代表已經訪問過的頂點的集合,v 可以代表當前所在的頂點為 v。S 代表的就是一種狀態(二進制表示), (11001)~2~ 代表在二進制中 {0,3,4} 三個頂點已經訪問過了,(11001)~2~ 代表的十進制數就是 25 ,所以當 S 為 25 的時候其實就是代表已經訪問過了 {0,3,4} 三個頂點,那假如一共有 5 個頂點(標號為 01234)話,所有的頂點都訪問完畢,Sj 就是 (11111)~2~。

二、常用位運算#

在這裡插入圖片描述
對於上面的一些操作舉例一些
就如之上的 (11001)~2~ 來說是已經訪問了 0,3,4 三個頂點,當我接下去要訪問頂點 2 的話可以有以下操作:1. 看看第 3 位是什麼?那麼選擇取出二進制數 x 的第 k 位,y = x >>(k-1) & 1,y = (110)~2~ & 1 = 0,2. 把第 3 位變為 1,y = x | (1<<(k-1)),y = (11001)~2~ | (100)~2~ = (11101)~2~,所以頂點 2 訪問完畢,加入到集合 S 中也結束了,此時 S 為 (11101) 2 = 29。

三、例題#

A-Hie with the Pie#

題意#

給出所有地點之間來往的所需時間(來往時間可能不同),如何用最短時間送完所有地點的訂單並返回披薩店(編號 0)

思路#

先用 Floyd 算法得出最短路矩陣,用 S 表示狀態,dp [S][i] 表示送達第 i 個地點所需的最短時間,由 S 可知當前已送達哪些城市,S = 1<<N-1 即為全部送達的狀態(N 位全為 1)

代碼#

// A-Hie with the Pie
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
#define div 1000000007
const int maxn = 12;
const int inf  = 0x3f3f3f;
int N,S,ans;
int dis[maxn][maxn];
int dp[1<<maxn][maxn];
void Floyd() {
    for(int k = 0; k <= N; ++k) {
        for(int i = 0; i <= N; ++i) {
            for(int j = 0; j <= N; ++j) {
                if(dis[i][j] > dis[i][k] + dis[k][j]) {
                    dis[i][j] = dis[i][k] + dis[k][j];
                }
            }
        }
    }
}
int main(){
    while(scanf("%d", &N) && N) {
        for(int i = 0; i <= N; ++i) {
            for(int j = 0; j <= N; ++j) {
                scanf("%d", &dis[i][j]);
            }
        }
        Floyd();
        for(int S = 0; S < (1<<N); ++S) {//S的所有狀態
            for(int i = 1; i <= N; ++i) {//所有要送的地點
                if(S & ( 1<<(i-1) )) {  //要送往地點i
                    if(S == ( 1<<(i-1) )) {//若S還沒有送往任何一個地點
                        dp[S][i] = dis[0][i];
                        break;
                    } else {
                        dp[S][i] = inf;
                        for(int j = 1; j <= N; ++j) {
                            if(j == i) continue;
                            //枚舉除當前要送往地點之外已送達地點,看時間是否更短
                            if(S&( 1<<(j-1) )) {
                                dp[S][i] = min(dp[S][i],dp[S^(1<<(i-1))][j] + dis[j][i]);
                            }
                            
                            
                        }
                    }
                }
            }
        }
        ans = inf;
        for(int i = 1; i <= N; ++i) {
            ans = min(ans, dp[(1<<N)-1][i] + dis[i][0]);
        }
        printf("%d\n",ans);
    }
    return 0;
}

B - Travelling#

題意#

給出 N 個城市、M 條道路成本,Acmer 先生想前往所有城市並且每個城市最多去兩次,求最低所需費用,若找不到這樣的路線則輸出 - 1

思路#

三進制狀壓 dp,每位上 0 表示沒去過,1 表示去過 1 次,2 表示去過 2 次,這裡用 cnt [i][j] 來表示 i 在 3 進制下從右往左第 j 位,用 three [i] 為三進制的第 i 位位權 即 three [i] 類似於二進制中 1<<i,最後若是全部訪問過且每個不超過 2 次則跟 ans 比較

代碼#

// B - Travelling
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
#define div 1000000007
const int maxn = 12;
const int maxt = 59049;
const int inf  = 0x3f3f3f;
int N,M,S,ans;
int three[maxn];//three[i]為三進制的第i位最大值 類似於二進制中1<<i
int map[maxn][maxn];
int cnt[maxt][maxn];//cnt[i][j]表示i在3進制下從右往左第j位
int dp[maxt][maxn];//dp[i][j]表示i狀態下前往j城市後總費用
void init() {
    memset(dp, -1, sizeof(dp));
    for(int i = 1; i <= N; ++i) {
        for(int j = 1; j <= N; ++j) {
            map[i][j] = inf;
        }
        map[i][i] = 0;
    }
}
int main(){
    int x = 3;
    three[0] = 0;
    three[1] = 1;
    for(int i = 2; i < maxn; ++i) {
        three[i] = x;
        x *= 3;
    }
    for(int i = 0; i < three[maxn-1]; ++i) {
        int now = i;
        for(int j = 1; j <= 10 && now; ++j) {
            cnt[i][j] = now % 3;
            now /= 3;
        }
    }
    while(~scanf("%d %d", &N, &M)) {
        init();
        for(int i = 1; i <= M; ++i) {
            int u,v,w;
            scanf("%d %d %d", &u, &v, &w);
            if(w < map[u][v]) {
                map[u][v] = map[v][u] = w;
            }
        }
        for(int i = 1; i <= N; ++i) {
            dp[three[i]][i] = 0;
        }
        int ans = inf;
        for(int S = 0; S < three[maxn-1]; ++S) {//S的所有狀態
            bool allvis = true;
            for(int i = 1; i <= N; ++i) {//枚舉每個城市
                if(cnt[S][i] == 0)   //該狀態下有城市還沒有被訪問過
                    allvis = false;
                if(dp[S][i] == -1) continue;
                for(int j = 1; j <= N; ++j) {   //枚舉下個能訪問的城市
                    if(i == j || map[i][j] == inf || cnt[S][j] >= 2) 
                        continue;   
                    int nS = S+three[j];//下個狀態
                    if(dp[nS][j] == -1 || dp[S][i] + map[i][j] < dp[nS][j]) { //下個城市的總費用小
                        dp[nS][j] = dp[S][i] + map[i][j];
                    }
                }
            }
            if(allvis) {
                for(int i = 1; i <= N; ++i) {
                    if(dp[S][i] != -1) {
                        ans = min(ans, dp[S][i]);
                    }
                }
            }
        }
        if(ans == inf) printf("-1\n");
        else printf("%d\n",ans);
    }
    return 0;
}
載入中......
此文章數據所有權由區塊鏈加密技術和智能合約保障僅歸創作者所有。