luogu P3478 [POI2008]STA-Station

题面传送门
明显是换根\(dp\),转移时讨论一下就好了。
代码实现:

#include<cstdio>
#include<cstring>
#define max(a,b) ((a)>(b)?(a):(b))
using namespace std;
int n,m,k,head,h[1000039],x,y,s[1000039];
long long ans,dp[1000039];
struct yyy{
	int to,z;
}f[2000039];
inline void add(int x,int y){
	f[++head]=(yyy){y,h[x]};
	h[x]=head;
}
inline void dfs1(int x,int last){
	int cur=h[x];
	yyy tmp;
	while(cur!=-1){
		tmp=f[cur];
		if(tmp.to!=last) dfs1(tmp.to,x),s[x]+=s[tmp.to],dp[x]+=dp[tmp.to];
		cur=tmp.z;
	}
	dp[x]+=s[x];
	s[x]++;
}
inline void dfs2(int x,int last){
	int cur=h[x];
	yyy tmp;
	while(cur!=-1){
		tmp=f[cur];
		if(tmp.to!=last) dp[tmp.to]=dp[x]-s[tmp.to]+n-s[tmp.to],dfs2(tmp.to,x); 
		cur=tmp.z;
	}
}
inline void read(int &x){
	char s=getchar();x=0;
	while(s<'0'||s>'9') s=getchar();
	while(s>='0'&&s<='9') x=(x<<3)+(x<<1)+(s^48),s=getchar();
}
int main(){
	memset(h,-1,sizeof(h));
	register int i;
	read(n);
	for(i=1;i<n;i++) read(x),read(y),add(x,y),add(y,x);
	dfs1(1,0);
	dfs2(1,0);
	for(i=1;i<=n;i++) ans=max(ans,dp[i]);
	for(i=1;i<=n;i++) if(dp[i]==ans){printf("%d\n",i);return 0;}
}