開始看這道題其實就是一個旅行商問題具體求法都差不多:抽出關鍵點,找到最短距離,然後狀態壓縮,這道題要注意的是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; }