题目:HDU - 4284 Travel
这题感觉很难,看了大佬的题解才发现,光是最短路还不够,还得用状压dp。先用弗洛伊德算法求出来每个城市之间所要的最少的路费。然后,再用状压dp来枚举所有的点,只有打工或不打工两种状态。然后看自己资金为大于等于0的情况下时候可以再回到城市1。
代码如下:
#include<stdio.h>
#include<iostream>
#include<map>
#include<algorithm>
#include<string.h>
#include<queue>
using namespace std;
typedef long long ll;
int n,m,s;
int pre[200][200];
int dp[1<<16][20];
int mp[200];
int p[200][2];
int main(){
ios::sync_with_stdio(false);
cin.tie();
cout.tie();
int t;
cin>>t;
while(t--){
memset(pre,0x3f3f3f3f,sizeof pre);
memset(dp,-1,sizeof dp);
cin>>n>>m>>s;
for(int i=1;i<=m;i++){
int l,r,ww;
cin>>l>>r>>ww;
if(ww<pre[l][r]){
pre[l][r]=pre[r][l]=ww;
}
}
for(int i=1;i<=n;i++){
pre[i][i]=0;
}
for(int k=1;k<=n;k++){
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(pre[i][j]>pre[i][k]+pre[k][j]){
pre[i][j]=pre[i][k]+pre[k][j];
}
}
}
}
int h;
cin>>h;
memset(mp,0,sizeof mp);
for(int i=1;i<=h;i++){
cin>>mp[i];
cin>>p[i][1];
cin>>p[i][0];
}
for(int i=1;i<=h;i++){
if(s-pre[1][mp[i]]-p[i][0]<0)continue;
dp[1<<(i-1)][i]=s-p[i][0]+p[i][1]-pre[1][mp[i]];
}
dp[0][1]=s;
for(int i=0;i<(1<<h);i++){
for(int j=1;j<=h;j++){
if(!(i&(1<<(j-1))))continue;
if(mp[j]==0)continue;
for(int k=1;k<=h;k++){
if(mp[k]==0)continue;
if(j==k||!(i&(1<<(k-1))))continue;
if(pre[mp[k]][mp[j]]==0x3f3f3f3f)continue;
if(dp[i^(1<<(j-1))][k]-pre[mp[k]][mp[j]]-p[j][0]<0)continue;
dp[i][j]=max(dp[i][j],dp[i^(1<<(j-1))][k]-pre[mp[k]][mp[j]]-p[j][0]+p[j][1]);
}
}
}
bool f=0;
for(int i=1;i<=h;i++){
if(dp[(1<<h)-1][i]-pre[mp[i]][1]>=0){
f=1;
break;
}
}
if(f==1)cout<<"YES\n";
else cout<<"NO\n";
}
}