HDU 4284--Travel（状压dp+TSP）

来源：http://acm.hdu.edu.cn/showproblem.php?pid=4284

题意：给出一些城市，从1出发，旅游一圈回到1，由于花费可能不够，所以选择一些城市打工，打工之前需要花费d买一个证，工资为c。选中的城市必须去工作一次，而且只能工作一次，问能不能完成旅行。

分析：先用floyd处理下，然后就是红果果的TSP了。

#include<iostream>
#include<algorithm>
#include<iomanip>
#include<cstring>
#include<string>
#include<cstdio>
#include<cmath>
#include<queue>
#include<set>
using namespace std;
#define inf 0x3f3f3f3f
int dp[1<<15][20];
int map[105][105],val[20],cost[20],must[20];
int n,v,m,h;
void floyd()
{
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(map[i][j]>map[i][k]+map[k][j])
                    map[i][j]=map[i][k]+map[k][j];
            }
        }
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&v);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                map[i][j]=inf;
            }
            map[i][i]=0;
        }
        for(int i=0;i<m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            if(map[x][y]>z)
                map[x][y]=map[y][x]=z;
        }
        floyd();
        scanf("%d",&h);
        for(int i=0;i<h;i++)
        {
            scanf("%d%d%d",&must[i],&val[i],&cost[i]);
        }
        memset(dp,inf,sizeof(dp));
        for(int i=0;i<h;i++)
        {
            if(v-map[1][must[i]]>=cost[i])
                dp[1<<i][i]=map[1][must[i]]-val[i]+cost[i];
        }
        for(int i=0;i<(1<<h);i++)
        {
            for(int j=0;j<h;j++)
            {
                if(dp[i][j]!=inf)
                {
                    for(int k=0;k<h;k++)
                    {
                        if(j!=k&&(i&(1<<k))==0)
                        {
                            int st=i|(1<<k);
                            if(v-dp[i][j]-map[must[j]][must[k]]>=cost[k])
                                dp[st][k]=min(dp[st][k],dp[i][j]+map[must[j]][must[k]]-val[k]+cost[k]);
                        }
                    }
                }
            }
        }
        int ans=inf;
        for(int i=0;i<h;i++)
        {
            ans=min(ans,dp[(1<<h)-1][i]+map[must[i]][1]);
        }
        printf("%s\n",ans<=v?"YES":"NO");
    }
	return 0;
}