NC54148货物收集

牛客难度：2
题目链接货物运输
解法1：DFS+sort
因为给出的是一棵树，可以先进行一边DFS求出到每个点的最大武力值，然后把所有点按到达需要的武力值进行升序排序。贪心，累加每个点的货物值直到sum超过W，此时点的达到需要的武力值就是答案。
解法2：二分+DFS
二分枚举答案，从1-1e9，然后DFS验证；
只放了解法2的代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
long long int head[maxn],a[maxn];
typedef long long int ll;
struct edge
{
	ll to,next,l;
}e[maxn*2];
ll cnt,n,w,tot;
void add(ll x,ll y,ll l)
{
	e[cnt].to=y;
	e[cnt].l=l;
	e[cnt].next=head[x];
	head[x]=cnt++;
}
void dfs(ll u,ll fa,ll k)
{
	tot+=a[u];
	for(ll i=head[u];i!=-1;i=e[i].next)
	{
		ll v=e[i].to;
		if(v==fa)continue;
		if(e[i].l>k)continue;
		dfs(v,u,k);
	}
}
ll solve()
{
	ll l=0,ans=0,r=1e9;
	while(l<=r)
	{
		ll mid=(l+r)>>1;
        tot=0;
		dfs(1,1,mid);
		if(tot>=w)
		{
			ans=mid;
			r=mid-1;
		}
		else l=mid+1;
	}
	return ans;
}
int main()
{
	ll x,y,z,ma=0;
	scanf("%d%d",&n,&w);
	memset(head,-1,sizeof(head));
	for(int i=2;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	ll ans=solve();
	printf("%lld",ans);
	return 0;
}