Educational Codeforces Round 67 (Rated for Div. 2) E. Tree Painting（树形dp）

题目链接



题意：要你求以每个点为根时，所有子树和的最大值。
思路：可以先dfs第一遍求出1为根时的答案，那么dp【v】=dp【u】+n-2*size【v】，（u是当前节点，v是u的儿子节点，size【v】是v的子树大小)

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+1;
typedef long long ll;
vector<int>g[maxn];
int n,u,v;
ll size[maxn],ans;
void dfs(int x,int fa)
{
	size[x]=1;
	for(int to:g[x])
	{
		if(to==fa) continue;
		dfs(to,x);
		size[x]+=size[to];
	}
	ans+=size[x];
}
void dfs1(int x,int fa,ll sum)
{
	ans=max(ans,sum);
	for(int to:g[x])
	{
		if(to==fa) continue;
	dfs1(to,x,sum+n-2*size[to]);
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;++i)
	scanf("%d%d",&u,&v),g[u].push_back(v),g[v].push_back(u);
	dfs(1,0);
	dfs1(1,0,ans);
	printf("%lld\n",ans);
}