hdu 4640 Island and study-sister(狀態壓縮dp)

先處理前兩個學長到達各個點所需要的最少時間,在計算前兩個學長和最後一個學長救出所有學妹的最少時間。

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
const int inf=1000000000;
struct node{
	int b,d;
};
struct node2{
    int dst,sta;
    node2(){}
    node2(int _dst,int _sta)
    {
        dst=_dst;sta=_sta;
    }
}q[3000000];
int dp[17][1<<17],n;
int f[17],dis[1<<17],dis2[1<<17],s;
vector<node>mp[17];
void bfs()//建立所有狀態
{
    int i,j,k,st=0,ed=0,sta,dst,tmp;
    node2 p;
    node u;
    p.dst=1;p.sta=1;
    dp[1][1]=0;
    q[ed++]=p;
    while(st<ed)
    {
        p=q[st];
        dst=p.dst;
        sta=p.sta;
        for(i=0;i<mp[dst].size();i++)
        {
            u=mp[dst][i];tmp=sta|f[u.b];

            if(dp[u.b][tmp]>dp[dst][sta]+u.d)
            {

                dp[u.b][tmp]=dp[dst][sta]+u.d;
                q[ed++]=node2(u.b,tmp);
            }
        }
        st++;
    }
}
int main()
{
	int t,i,j,m,k,a,b,d,h,iter,juli,v=1,x;
	f[1]=1;
	for(i=2;i<=17;i++)f[i]=f[i-1]<<1;
	node p;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);
		for(i=1;i<=n;i++)mp[i].clear();
		for(i=1;i<=n;i++)
        for(j=0;j<1<<n;j++)
           dp[i][j]=inf;
		while(m--)
		{
			scanf("%d%d%d",&a,&b,&d);
			p.b=b;p.d=d;
			mp[a].push_back(p);
			p.b=a;
			mp[b].push_back(p);
		}
		bfs();
		for(i=1;i<1<<n;i+=2){//求出所有狀態的最短時間
			dis[i]=inf;
		    for(j=1;j<=n;j++)
		    	if(dp[j][i]<dis[i])
		    		dis[i]=dp[j][i];
		}
		scanf("%d",&k);
		int tmp=0;
		for(i=0;i<k;i++)//目標狀態
        {
            scanf("%d",&s);
            tmp|=f[s];
        }
        printf("Case %d: ",v++);
        int ans=inf;
        for(i=0;i<1<<n;i++)dis2[i]=inf;//兩個學長一起
        for(i=0;i<1<<n;i++)
        for(h=j=i^((1<<n)-1);;j=h&(j-1)){
            if(dis2[i|j|1]>(x=max(dis[i|1],dis[j|1])))
                dis2[i|j|1]=x;
            if(j==0)break;
        }
        for(i=0;i<1<<n;i++)//三個一起
        for(h=j=i^((1<<n)-1);;j=h&(j-1)){
            if(( ( (i|j|1)&tmp  )==tmp) && ans>(x=max(dis[i|1],dis2[j|1])))
                ans=x;
            if(j==0)break;
        }
        if(ans==inf)ans=-1;
        printf("%d\n",ans);
	}
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章