最大权值子树

题目链接

好烦好烦好烦，看不到未来在哪里，这么简单的题都不会做，蓝桥这种水赛省三都拼尽全力无法战胜，我怎么这么笨

求最大权值子树，每个点都在子节点里选大于0的加上去，最后加上本身，再返回父节点

void dfs(int now)
{
	res[now]=value[now];
	for(int i=head[now];i;i=edge[i].nxt)
	{
		int to=edge[i].to;
		if(vis[to]) continue;
		vis[to]=1;
		dfs(to);
		res[now]+=max(res[to],0ll);
	}
	ans=max(ans,res[now]);
}

不一定根节点就是最大，max每个点

允许不选任何节点，最大值可能是负数，用max

#include <bits/stdc++.h>
using namespace std;
typedef long long int LL;
const int N=1e5+10,MOD=1e9+7;
LL n,ans,cnt,head[N],res[N],vis[N],value[N];
struct Edge{
	int to=0,nxt=0;
}edge[N<<1];

void add(int f,int t)
{
	edge[++cnt].nxt=head[f];
	edge[cnt].to=t;
	head[f]=cnt;
}

void dfs(int now)
{
	res[now]=value[now];
	for(int i=head[now];i;i=edge[i].nxt)
	{
		int to=edge[i].to;
		if(vis[to]) continue;
		vis[to]=1;
		dfs(to);
		res[now]+=max(res[to],0ll);
	}
	ans=max(ans,res[now]);
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++) cin>>value[i];
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		cin>>x>>y;
		add(x,y),add(y,x);
	}
	vis[1]=1;
	dfs(1);
	
	cout<<max(0ll,ans);
	return 0;
}