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;

	}
}