[nk] 牛客月赛45 E.筑巢

前言

传送门 :

思路

标签 : 树形$dp$

状态表示 : $f[i]$, 以$i$为树根的最大舒适度
状态计算 : $f[i]=max(f[i],f[j]+w[i]+f[i])$

记得开$ll$

Mycode

void dfs(int u){
	if(st[u]) return;
	st[u] = 1;
	f[u] += a[u];
	
	for(auto x : g[u]){
		if(st[x.to])  continue;
		dfs(x.to);
		f[u] =  max(f[u],f[u]+f[x.to]+x.val);
	}
}

void solve()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	cin>>a[i];
	
	for(int i=1;i<n;i++){
		int a,b,c;cin>>a>>b>>c;
		g[a].pb({b,c});
		g[b].pb({a,c});
	}
	
	
	for(int i=1;i<=n;i++){
		if(!st[i])
		dfs(i);
	}
	
	for(int i=1;i<=n;i++){
		ans = max(ans,f[i]);
	}
	cout<<ans<<endl;
}