[题解] LuoguP5666 树的重心

这个题......确实是CSPNOIP题qwq

感觉猜到一个性质就差不多了,首先,对于一棵树,随便拎一个节点\(rt\)当根节点,那么他的重心一定在\(rt\)的重儿子里,进一步的,可以发现重心一定在\(rt\)向下的重链上。

根据这条性质,以及重链上从上到下节点的\(size\)一定是递减的,所以考虑在重链上倍增。

具体的令\(f[x][i]\)表示\(rt\)为根的时候,节点\(x\)沿着重链向下走\(2^i\)步到达的节点,这样,求重心从大到小枚举\(i\),向下跳就好了,另外一些细节可以自己画一个简单的图感性理解一下\(qwq\)。

基于上面的那个求重心的方法,类似换根\(dp\),二次扫描维护\(f\)和重儿子求答案就好了。

实现详见代码(配合c++11食用qaq):

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef long long ll;
typedef double db;
typedef pair<int,int> PII;
typedef vector<int> VI;

const int N=3e5+10,LOGN=20;
VI g[N];
int n,siz[N],f[N][23],son[N],fa[N];
ll ans;

void getf(int x) {
    rep(i,1,LOGN) f[x][i]=f[f[x][i-1]][i-1];
}
void calc(int x) {
    int u=x; 
    per(i,0,LOGN) {
        if(f[u][i]&&siz[f[u][i]]*2>=siz[x]) // 这里可以画张图感性理解一下qwq
            u=f[u][i];
    }
    if(siz[u]*2==siz[x]) ans+=fa[u];
    ans+=u;
}

void dfs(int x,int ff) {
    fa[x]=ff;
    siz[x]=1,siz[son[x]=0]=0;
    for(auto v:g[x]) if(v!=ff) {
        dfs(v,x),siz[x]+=siz[v];
        if(siz[v]>siz[son[x]]) son[x]=v;
    }
    f[x][0]=son[x],getf(x);
}

void getans(int x,int ff) {
    int x1=0,x2=0; siz[0]=0;
    for(auto v:g[x]) { // 枚举与x相邻的点,算出最大的儿子和次大的儿子
        if(siz[v]>=siz[x1]) x2=x1,x1=v;
        else if(siz[v]>=siz[x2]) x2=v;
    }
    for(auto v:g[x]) if(v!=ff) {
        calc(v);
        f[x][0]=v==x1?x2:x1,getf(x); // 
        siz[x]-=siz[v],siz[v]+=siz[x];
        calc(x);
        fa[x]=v,getans(v,x);
        siz[v]-=siz[x],siz[x]+=siz[v]; // 在算完边(x,v)对答案的贡献的时候要注意撤回影响
    }
    f[x][0]=son[x]; // 回溯的时候撤销操作
    getf(x),fa[x]=ff;
}

void sol() {
    memset(siz,0,sizeof(siz));
    memset(son,0,sizeof(son));
    memset(f,0,sizeof(f));
    memset(fa,0,sizeof(fa)); // 多测不清空抱凛两行泪TAT
    ans=0; scanf("%d",&n);
    rep(i,0,n+1) g[i].clear();
    rep(i,0,n-1) {
        int x,y; scanf("%d%d",&x,&y);
        g[x].pb(y),g[y].pb(x);
    }
    dfs(1,0),getans(1,0);
    printf("%lld\n",ans);
}

int main() {
    int _; scanf("%d",&_); while(_--) sol();
    return 0;
}

另外,听说NOIP复活了?

上一篇:省选测试8


下一篇:IPSec实验配置