筑巢（dfs）

牛客小白月赛

dfs直接搜索就可以啦

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int cnt=0;
const int N=1e5+10;
long long  w[2*N],e[2*N],f[2*N],nx[2*N],a[N],st[N];
long long  dp[N];
void add(int x,int y,int z)
{
	w[cnt]=z;
	e[cnt]=y,nx[cnt]=f[x],f[x]=cnt++;
}
void dfs(int x)
{
	if(st[x])
		return ;
	st[x]=1;
	dp[x]=a[x];
	int i;
	for(i=f[x]; ~i; i=nx[i])
	{
		int j=e[i];
		if(st[j])
			continue;
		dfs(j);
//		st[j]=1;
		if(dp[j]+w[i]>0)
		{
			dp[x]+=dp[j]+w[i];
		}

	}
}
int main()
{
	int i,j,k,m,n;
	long long x,y,z;
	scanf("%d",&n);
	for(i=1; i<=n; i++)
	{
		scanf("%lld",&a[i]);
	}
	memset(f,-1,sizeof(f));
	for(i=1; i<n; i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);//无向边
	}
	for(i=1; i<=n; i++)
	{
		if(!st[i])//防止重复
			dfs(i);
	}
	long long ans=-0x3f3f;
	for(i=1; i<=n; i++)
	{
		ans=max(ans,dp[i]);//比较哪个最舒适
	}
	printf("%lld\n",ans);
	return 0;
}