$重心定義:去掉重心后,樹分成若干個(gè)連通塊,要使最大連通塊最小$

$考慮x為根的樹和x的一些兒子y$

$現(xiàn)在已知y為根的子樹的重心s,如何遞推到x為根的子樹?$

$這么想,y子樹?>s子樹的過程中,也就是在y的頭上加了x和x的其他兒子$

$也就是如果把s砍掉,頭上的那個(gè)連通塊比以前大$

假如頭上那個(gè)連通塊不是最大的,那位置s為重心一定最優(yōu)

假如頭上的連通塊變成最大的,那把s往上跳,最大連通塊會(huì)變小,答案會(huì)變優(yōu)

所以每次嘗試往上跳就行了

#include <bits/stdc++.h> using namespace std; const int maxn=6e5+10; int rt[maxn],sz[maxn],mx[maxn],fa[maxn],n,m; struct p{int to,nxt; }d[maxn]; int head[maxn],cnt=1; void add(int u,int v){d[cnt]=(p){v,head[u]},head[u]=cnt++; } void dfs(int now,int pre) {sz[now]=1;for(int i=head[now];i;i=d[i].nxt){dfs( d[i].to,now );sz[now]+=sz[d[i].to];mx[now]=max(mx[now],sz[d[i].to]);}int me=mx[now];rt[now]=now;for(int i=head[now];i;i=d[i].nxt){int it=rt[ d[i].to ];int nice=max( mx[it],sz[now]-sz[it] ),k=it;while( fa[it]!=now ){it=fa[it];//往上跳int nn=max( mx[it],sz[now]-sz[it]);if( nn<nice ) nice=nn,k=it;else break;//不會(huì)更優(yōu)秀了 }if( me>nice ){me=nice;rt[now]=k;}} } int main() {cin >> n >> m;for(int i=2;i<=n;i++){cin >> fa[i];add(fa[i],i);}dfs(1,0);while( m-- ){int x; cin >> x;cout << rt[x] << '\n';}return 0; }

總結(jié)

以上是生活随笔為你收集整理的B. Kay and Snowflake(重心的性质)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。