换根DP模板

给你一个无根树,问你以哪个节点为根节点的时候得到所有点的深度之和最大
《贴一张 知乎大佬的一个解释》

#include<bits/stdc++.h>
using namespace std;
const int N = 2e6+10;
using ll = long long;
ll dep[N],sz[N];
ll ans[N];
vector<int>g[N];
int n;void dfs(int u,int father){sz[u] = 1;dep[u] = dep[father]+1;for(auto &t:g[u]){if(t==father)continue;dfs(t,u);sz[u]+=sz[t];}
}void down(int u,int father){for(auto &t:g[u]){if(t==father)continue;ans[t] = ans[u]-2*sz[t]+sz[1];down(t,u);}
}int main()
{cin>>n;for(int i=1;i<n;i++){int a,b;cin>>a>>b;g[a].push_back(b);g[b].push_back(a);}dfs(1,0);for(int i=1;i<=n;i++)ans[1]+=dep[i];//for(int i=1;i<=n;i++)cout<<sz[i]<<"\n";down(1,-1);ll res = ans[1],p = 1;for(int i=1;i<=n;i++)if(ans[i]>res)res=ans[i],p=i;cout<<p;
}