开始看这道题其实就是一个旅行商问题具体求法都差不多:抽出关键点,找到最短距离,然后状态压缩,这道题要注意的是1.起点可能没有包含在必须去的点里 2 在必须去的点中如果能打工 则必须打工赚钱(感觉题意不是这样的 不过只能这样了) 注意初始化dp是可能打工也可能不打工。
#include<stdio.h> #include<string.h> #include<iostream> #include<map>; using namespace std; #define INF 0x3f3f3f3f int dp[1<<16][16]; int Map[110][110]; struct Node { int ii; int pay; int get; }point[20]; int swap(int &a,int &b) { int t; t=a; a=b; b=t; return 0; } int min(int a,int b) { return a<b?a:b; } int main() { int n,m,i,j,T,money; scanf("%d",&T); while(T--) { scanf("%d%d%d",&n,&m,&money); memset(Map,INF,sizeof(Map)); for(i=1;i<=m;i++) { int a,b,c; scanf("%d%d%d",&a,&b,&c); Map[a][b]=Map[b][a]=min(Map[a][b],c); } for(int k=1;k<=n;k++) for(i=1;i<=n;i++) for(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 k; scanf("%d",&k); for(i=0;i<k;i++) { scanf("%d%d%d",&point[i].ii,&point[i].get,&point[i].pay); if(point[i].ii==1) { swap(point[i].ii,point[0].ii); swap(point[i].pay,point[0].pay); swap(point[i].get,point[0].get); } } if(point[0].ii!=1) { point[k].ii=1; point[k].pay=0; point[k].get=0; swap(point[k].ii,point[0].ii); swap(point[k].pay,point[0].pay); swap(point[k].get,point[0].get); k++; } memset(dp,-1,sizeof(dp)); if(money>=point[0].pay) dp[1][0]=money-point[0].pay+point[0].get; else dp[1][0]=money; int t=(1<<k); for(i=1;i<t;i++) { for(j=0;j<k;j++) { if((i&(1<<j))==0) continue; if(dp[i][j]<0) continue; for(int x=0;x<k;x++) { if(x==j) continue; if((i&(1<<x))!=0) continue; if(dp[i][j]>=Map[point[j].ii][point[x].ii]+point[x].pay) { dp[i|(1<<x)][x]=max(dp[i|(1<<x)][x],dp[i][j]+point[x].get-Map[point[j].ii][point[x].ii]-point[x].pay); } } } } int flash=0; for(i=0;i<k;i++) { if(dp[t-1][i]>=Map[point[i].ii][1]) flash=1; } if(flash) printf("YES\n"); else printf("NO\n"); } return 0; }