hdu 4114 Disney's FastPass 状压dp

点击打开链接

题意：

游戏园里有N个区域，有M条边连接这N个区域，有K个要访问的景点。对于每个景点告诉你这个景点所在的区域，要访问这个景点需要等待一定时间，如果没有FastPass，等待时间有Ti，否则等待时间为FTi，接下来的Ni，表示有Ni个区域可以得到这个景点的FastPass，问从区域1出发，再回到区域1所需要的最少时间。

思路：

状态压缩，dis[i][s1][s2]，表示当前走到这个点，s1表示拿到了哪些景点的FastPass，s2表示已经访问了哪几个景点。

但是需要注意的是：当游览过某个景点之后，就认为拿着了这个景点的票了！！！这个优化很重要

代码：

#include <bits/stdc++.h>
using namespace std;

int n,m,k;
int dis[55][55],dp[55][1<<8][1<<8],pos[10],t[10],ft[10],pas[55];
int ans;

void init(){
    ans = 0x3f3f3f3f;
    memset(dp,0x3f,sizeof(dp));
    memset(dis,0x3f,sizeof(dis));
    memset(pas,0,sizeof(pas));
}

void floyd(){
    for(int k=1; k<=n; k++)
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
                dis[i][j] = min(dis[i][j],dis[i][k]+dis[k][j]);
}

void solve(){
    dp[1][0][0] = 0;
    for(int s1=0; s1<(1<<k); s1++){
        for(int s2=0; s2<(1<<k); s2++){
            for(int i=1; i<=n; i++){
                int now = dp[i][s1][s2];
                if(now==0x3f3f3f3f) continue;
                if(s2==(1<<k)-1) ans = min(ans,now+dis[i][1]);
                for(int j=0; j<k; j++){
                    if((s2&(1<<j)) == 0){ // 没走过第j个景点
                        int& nxt = dp[pos[j]][s1|pas[pos[j]]][s2^(1<<j)]; // 当游览过某个景点之后，就认为拿着了这个景点的票了！！！
                        int add = dis[i][pos[j]];
                        if(s1 & (1<<j)) add += ft[j];
                        else add += t[j];
                        nxt = min(nxt,now+add);
                    }
                }
                for(int j=1; j<=n; j++){  // 中转 通过j得到fastpass
                    int& nxt = dp[j][s1|pas[j]][s2];
                    int add = dis[i][j];
                    nxt = min(nxt,now+add);
                }
            }
        }
    }
}

int main(){
    int T; cin>>T;
    for(int cas=1; cas<=T; cas++){
        
        init();
        cin >> n >> m >> k;
        for(int i=1; i<=m; i++){
            int u,v,w; cin>>u>>v>>w;
            dis[u][v] = w; dis[v][u] = w;
        }
        for(int i=0;i<n;i++) dis[i][i]=0; 
        floyd();

        for(int i=0; i<k; i++){
            cin >> pos[i] >> t[i] >> ft[i];
            int num; cin>>num;
            for(int j=0; j<num; j++){
                int tmp; cin>>tmp;
                pas[tmp] |= (1<<i);
            }
        }

        solve();

        cout << "Case #" << cas << ": " << ans << endl;

    }
}