2023河南萌新联赛第（一）场：河南农业大学 D - 松鼠回家

2023河南萌新联赛第（一）场：河南农业大学 D - 松鼠回家

题目描述

松鼠宝宝由于贪玩去了一个具有n个点和m条边的无向图中，现在松鼠宝宝仅有h点体力，所有的边经过一次后会消耗部分体力，同时松鼠爸爸为了惩罚贪玩的松鼠宝宝，每到一个点会扣除部分松果（起点的松果也会扣除）。现松鼠宝宝向你求助，询问在能到达家的情况下

尽可能让路径上扣除松果的数量最大的那个点扣除的数量尽可能小。

输入描述:

第一行读入五个数 $n，m，st，ed，h$（分别无向图的点数，边数，起点位置，家的位置，开始时候的体力）

接下来一行读入 $n$ 个数 $a_i$（每个点所扣除的松果数量）

接下来 $m$ 行读入 $x，y，z$ （分别代表无向边的两点和路上所消耗的体力）

$1<=n<=1e4$

$1<=m<=2e4$

$1<=a_i,z,h<=1e7$

$1<=x,y<=n$

输出描述:

输出一行代表最大扣除数量的最小值，若无法到达，则输出-1

示例1

输入

4 4 1 4 8
8
5
6
10
1 3 4
2 4 1
2 1 2
3 4 3

输出

10

#include
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int N = 10010, INF = 0x3f3f3f3f;
vector<PII> g[N];
bool vis[N];

int a[N];
int n,m,st,ed,h;
int ans=INF;

void dfs(int u,int fa,int sum,int mx)
{
	if(mx>ans || sum>h) return ;
	if(u==ed)
	{
		ans=min(ans,mx);
		return ;
	}
	for(auto &p:g[u])
	{
		int j=p.first,w=p.second;
		if(j==fa || vis[j]) continue;
		vis[j]=true;
		dfs(j,u,sum+w,max(mx,a[j]));
		vis[j]=false;
	}
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    cin>>n>>m>>st>>ed>>h;
    for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=m;i++)
	{
		int u,v,w;cin>>u>>v>>w;
		g[u].push_back({v,w});
		g[v].push_back({u,w});
	} 
	dfs(st,-1,0,a[st]);
	if(ans==INF) ans=-1;
	cout<<ans;
    
    return 0;
}