这个题我们可以建图再去搜索
当一条边连接的两个城市能够从一个到另一个能赚钱的话
比如说输入 x y z,如果x城市到y城市能赚钱(也就是price[y]>=price[x]+z)
就建立一条从x到y的单向边
那么DFS从所有入度为0的点开始沿着边搜索,势必钱是越赚越多的
所以更新赚的钱即可
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int N=100000+5;
int t,n,u,v,d,ans,num;
int val[N],in[N];
struct P
{
int to,cost;
};
vector<P> G[N];
void DFS(int id)
{
for(int i=0;i<G[id].size();i++){
P p=G[id][i];
num+=p.cost;
ans=max(val[p.to]-num,ans);//ans是赚的钱
DFS(p.to);
num-=p.cost;
}
}
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(int i=0;i<=n;i++) G[i].clear();
memset(in,0,sizeof(in));
ans=0;
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
for(int i=1;i<n;i++){
scanf("%d%d%d",&u,&v,&d);
if(val[u]>=val[v]+d){
G[v].push_back(P{u,d});
in[u]++;
}else if(val[v]>=val[u]+d){
G[u].push_back(P{v,d});
in[v]++;
}
}
for(int i=1;i<=n;i++){
if(!in[i]){
num=val[i];//num是花费
DFS(i);
}
}
printf("%d\n",ans);
}
return 0;
}
这个代码比较优秀(学弟写的),每输入一条边,就更新两个城市赚的钱(从x到y,从y到x的)
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int N=100000+5;
int t,n,u,v,d;
int val[N],dp[N];
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d",&n);
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
int ans=-1;
for(int i=1;i<n;i++){
scanf("%d%d%d",&u,&v,&d);
ans=max(dp[u]=max(dp[v]-d-val[v]+val[u],dp[u]),ans);
ans=max(dp[v]=max(dp[u]-d-val[u]+val[v],dp[v]),ans);
}
printf("%d\n",ans);
}
return 0;
}