hdu 4114 狀態壓縮+最短路

http://acm.hdu.edu.cn/showproblem.php?pid=4114

人渣不拆啊，寫了個spfa，T了，然後寫了個dij，MLE了。我呵呵呵，然後我以爲難道不是最短路？？？就看了下題解。。。發現是自己SB了。。。我真弱啊，真弱。。。。

對於這道題：很容易想到用狀態壓縮來表示某個狀態下的最短路，

dis[50][1<<8][1<<8]，dis[N][V][T]表示到N號節點，瀏覽過的點的狀態爲V，拿着的票的狀態爲T，時候的最短路程。

然後就是個最短路了，思路很簡單。

但是需要注意的是：當遊覽過某個景點之後，就認爲拿着了這個景點的票了！！！這個優化很重要。。。要不就是SPFA，TLE。。dij由於邊太多MLE。。。我還Dij寫SB了。。。真是太弱了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int maxn = 51;
const int INF = 0x3ffffff;
int n,m,k;
int pass[maxn];
int place[maxn];
int len[maxn][maxn];
int play[maxn][2];
int dis[maxn][1<<8][1<<8];
bool vis[maxn][1<<8][1<<8];

struct E{
	int s,w;
	void get(int &u,int &vi,int &ti,int &w){
		u=s/(1<<16);
		vi=s/(1<<8)%(1<<8);
		ti=s%(1<<8);
		w=this->w;
	}
	void set(int s,int w){
		this->s=s;
		this->w=w;
	}
	friend bool operator < (const E &a,const E &b){
		return a.w>b.w;
	}
};
int get_hash(int u,int vi,int ti){
	return u*(1<<16)+vi*(1<<8)+ti;
}
priority_queue<E>q;
void dij(){
	for(int i=1;i<=n;i++){
		for(int b1=0;b1<1<<k;b1++){
			for(int b2=0;b2<1<<k;b2++){
				dis[i][b1][b2]=INF;
				vis[i][b1][b2]=false;
			}
		}
	}
	while(!q.empty())q.pop();
	E tmp;
	tmp.set(get_hash(1,0,0),0);
	q.push(tmp);
	int u,vi,ti,w;
	while(!q.empty()){

		tmp=q.top();q.pop();
		tmp.get(u,vi,ti,w);
		if(vis[u][vi][ti])continue;

		dis[u][vi][ti]=w;
		vis[u][vi][ti]=true;
		
		if(u==1&&vi==(1<<k)-1&&ti==vi)return;
		
		int d=0;
		if((vi|place[u])!=vi&&!vis[u][vi|place[u]][ti]){
			for(int i=0;i<k;i++){
				if( (place[u]&(1<<i)) && !(vi&(1<<i)) ){
					if(ti & (1<<i) ){
						d+=play[i][1];
					}else d+=play[i][0];
				}
			}
			
			if( w+d<dis[u][vi|place[u]][ti]){
				dis[u][vi|place[u]][ti]=w+d;
				tmp.set(get_hash(u,vi|place[u],ti),w+d);
				q.push(tmp);
			}
		}

		if((ti|pass[u]|vi)!=ti){
			if(!vis[u][vi][ti|pass[u]|vi] && w<dis[u][vi][ti|pass[u]|vi]){
				dis[u][vi][ti|pass[u]|vi]=w;
				tmp.set(get_hash(u,vi,ti|pass[u]|vi),w);
				q.push(tmp);
			}
		}

		for(int i=1;i<=n;i++){
			if(!vis[i][vi][ti|vi] && w+len[u][i] < dis[i][vi][ti|vi]){
				dis[i][vi][ti|vi]=w+len[u][i];
				tmp.set(get_hash(i,vi,ti|vi),w+len[u][i]);
				q.push(tmp);
			}
		}
		
	}
}

int main(){
	int T,ca=1;
	scanf("%d",&T);
	while(T--){
		printf("Case #%d: ",ca++);
		scanf("%d%d%d",&n,&m,&k);
		memset(pass,0,sizeof(pass));
		memset(place,0,sizeof(place));
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				len[i][j]=i==j?0:INF;
			}
		}
		for(int i=0;i<m;i++){
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			len[u][v]=len[v][u]=min(len[u][v],w);
		}
		for(int i=0;i<k;i++){
			int p;
			scanf("%d%d%d",&p,&play[i][0],&play[i][1]);		
			place[p]=place[p]|(1<<i);
			int nt;
			scanf("%d",&nt);
			for(int j=0;j<nt;j++){
				scanf("%d",&p);
				pass[p]|=1<<i;
			}
		}
		dij();
		printf("%d\n",dis[1][(1<<k)-1][(1<<k)-1]);
	}
}