题目传送门
题意: 有一颗树,问你选哪个节点为根节点时,所有节点深度之和最大?
思路: 先以1为根,求出每个点的深度和以该节点为根的子树大小。 f [ x ] f[x] f[x]表示以x为根节点时,所有节点的深度之和,再深搜一遍,如果y是x的子节点,那么 f [ y ] = f [ x ] − s i z [ y ] + ( n − s i z [ y ] ) f[y]=f[x]-siz[y]+(n-siz[y]) f[y]=f[x]−siz[y]+(n−siz[y]),即从x作为根节点转化成从y作为根节点,y的子树中所有点的深度-1,其他所有点深度+1,从而得到了这个状态转移式。
代码:
#include<bits/stdc++.h>
#define endl '\n'
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define pii pair<int,int>
#define sz(x) (int)(x).size()
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read()
{
int x=0,f=1;
char ch=gc();
while(ch<'0'||ch>'9')
{
if(ch=='-')
f=-1;
ch=gc();
}
while(ch>='0'&&ch<='9')
{
x=x*10+ch-'0';
ch=gc();
}
return x*f;
}
using namespace std;
const int N=1e6+10;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-7;
vector<int>e[N];
int siz[N],f[N],dep[N],n;
void dfs(int fa,int x,int d)
{
siz[x] = 1;
dep[x] = d;
for(auto i:e[x])
{
if(i!=fa)
{
dfs(x,i,d+1);
siz[x] += siz[i];
}
}
}
void ddfs(int fa,int x)
{
for(auto i:e[x])
{
if(i!=fa)
{
f[i] = f[x] - siz[i] + (n-siz[i]);
ddfs(x,i);
}
}
}
void solve()
{
cin>>n;
for(int i=1;i<=n-1;i++)
{
int u,v;
cin>>u>>v;
e[u].pb(v);
e[v].pb(u);
}
dfs(1,1,1);
for(int i=1;i<=n;i++)
{
f[1] += dep[i];
}
ddfs(1,1);
int res,ma=-1;
for(int i=1;i<=n;i++)
{
if(f[i]>ma)
{
ma=f[i];
res=i;
}
}
cout<<res<<endl;
}
signed main()
{
solve();
return 0;
}