【状压DP】 HDU 4284 Travel

点击打开链接

题意：从点1开始旅行N个地区，每个地区之间需要路费，要求必须经过特定的H个地区，因为钱不够所以要在这些地区打工（只能一次），但是先要购买该地区的工作证，钱不够不能购买。

因为有N个地区(N>=H) ,可以不管其他地区，用最短路得到H地区之间的最短路，然后就是一般的状压dp

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <deque>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
#define FOR(i,a,b)  for(int i=a;i<=b;i++)
#define IN   freopen ("in.txt" , "r" , stdin);
#define OUT  freopen ("out.txt" , "w" , stdout);
typedef long long  LL;
const int MAXN = 10052;
const int MAXM = 6000010;
const int INF = 0x3f3f3f3f;
const LL mod = 10e9+7;
const double eps= 1e-8;
#define lson l,m, rt<<1
#define rson m+1,r,rt<<1|1
int c[155],d[155],must[155],flag;
int dp[(1<<15)+2][16],mp[122][122];
void init()
{
    flag=0;
    cler(dp,-1);cler(mp,INF);
    for(int i=0;i<100;i++) mp[i][i]=0;
}
int n,m,t,now,h;
int main()
{
    cin>>t;
    while(t--)
    {
        cin>>n>>m>>now;
        int u,v,w;
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            u--,v--;
            if(w<mp[u][v])
                mp[u][v]=mp[v][u]=w;
        }
        for(int k=0;k<n;k++)//floyd
            for(int i=0;i<n;i++)
            {
                if(mp[k][i]==INF) continue;
                for(int j=0;j<n;j++)
                {
                    if(mp[k][j]==INF) continue;
                    mp[i][j]=min(mp[i][k]+mp[k][j],mp[i][j]);
                }
            }
        cin>>h;
        for(int i=0;i<h;i++)
        {
            cin>>must[i]>>c[i]>>d[i];
            must[i]--;
        }
        for(int i=0;i<h;i++)//先处理起点出发的点
        {
            if(now-d[i]-mp[must[i]][0]>=0)
                dp[1<<i][i]=now-d[i]-mp[must[i]][0]+c[i];
        }
        int mx=(1<<h);
        for(int i=0;i<mx;i++)
        {
            for(int j=0;j<h;j++)
            {
                if(i&(1<<j)==0||dp[i][j]==-1) continue;
                for(int k=0;k<h;k++)
                {
                    if(i&(1<<k)) continue;
                    if(dp[i][j]-mp[must[j]][must[k]]-d[k]>=0)
                        dp[i|(1<<k)][k]=max(dp[i][j]-mp[must[j]][must[k]]-d[k]+c[k],dp[i|(1<<k)][k]);
                }
            }
        }
        for(int i=0;i<h;i++)//判断能否回到起点
        {
            if(dp[mx-1][i]-mp[must[i]][0]>=0)
            {
                flag=1;
                break;
            }
        }
        if(flag) puts("YES");
        else puts("NO");
    }
    return 0;
}