CF1187E Tree Painting 换根dp

题目传送门

这题一看就很换根dp（

换根dp其实挺有意思的。这题只需在草稿纸上画画就能推出式子了。

记得开longlong！！

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,head[205000],cnt,size[205000],f[205000],s[205000];
int ans;
struct edge{
	int v,next;
}e[405000];
void add(int u,int v){
	e[++cnt].v=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}
void dfs1(int u,int fa){
	size[u]=1;
	for(int i=head[u];i;i=e[i].next)
		if(e[i].v!=fa){dfs1(e[i].v,u);size[u]+=size[e[i].v];f[u]+=f[e[i].v];}
	f[u]+=size[u];
}
void dfs2(int u,int fa){
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].v;
		if(e[i].v!=fa){
			s[v]=n-2*size[v]+s[u];
			dfs2(v,u);
		}
  }
}
signed main(){
	scanf("%lld",&n);
	for(int i=1;i<=n-1;++i){
		int u,v;
		scanf("%lld%lld",&u,&v);
		add(u,v);add(v,u);
	}
	dfs1(1,0);
	s[1]=f[1];
	dfs2(1,0);
	for(int i=1;i<=n;++i)ans=max(ans,s[i]);
	printf("%lld\n",ans);
	return 0;
}