洛谷 P3478 [POI2008]STA-Station 【换根DP】

题目链接
题意：

找一个树上的结点，使其他节点到它的距离和最大

题解：

换根DP模板题，每次将根节点和它的子节点调换统计答案

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<algorithm>
#include<cstdio>
#include<list>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;


#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}

const int INF = 0x3f3f3f3f;
const int N = 1e6+10;
int n;
int f[N],sz[N],dep[N];
struct edge{
	int to,next;
}e[N*2];
int head[N*2],cnt=0;
void add(int u,int v){
	e[++cnt]={v,head[u]},head[u]=cnt;
	e[++cnt]={u,head[v]},head[v]=cnt;
}
void dfs(int u,int fa){
	sz[u]=1,dep[u]=dep[fa]+1;
	for(int i=head[u];i;i=e[i].next){
		if(e[i].to==fa)continue;
		dfs(e[i].to,u);
		sz[u]+=sz[e[i].to];
	}
}
void dfs2(int u,int fa){
	for(int i=head[u];i;i=e[i].next){
		if(e[i].to==fa)continue;
		f[e[i].to]=f[u]+n-2*sz[e[i].to];
		dfs2(e[i].to,u);
	}
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<n;i++){
		int u,v; cin>>u>>v;
		add(u,v);
	}
	dfs(1,1);
	for(int i=1;i<=n;i++)f[1]+=dep[i];
	dfs2(1,1);
	int ans=0,p;
	for(int i=1;i<=n;i++)if(f[i]>ans)ans=f[i],p=i;
	cout<<p<<endl;
}