[BZOJ2152]聪聪可可

本文探讨了点分治算法在解决图论问题中的应用,通过具体实例展示了如何利用该算法来优化求解过程,提高了算法的效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

原题地址

点分治.

AC code:

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
const int N=20010;
const int INF=1<<28;
int  n,a,m,mins,root;
int  son[N];
int  f[N][3];
bool del[N];

struct Data{
    int v,w;

    Data(int v,int w):v(v),w(w) {}
};
vector<Data> G[N];

void read(){
    scanf("%d",&n);
    a=n;
    for(int i=1;i<n;i++){
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        G[x].push_back(Data(y,w));
        G[y].push_back(Data(x,w));
    }
}

void DFS1(int pre,int p){
    int maxs=-INF;
    son[p]=1;
    for(int i=0;i<G[p].size();i++){
        Data q=G[p][i];
        if(q.v==pre||del[q.v]) continue;
        DFS1(p,q.v);
        son[p]+=son[q.v];
        maxs=max(maxs,son[q.v]);
    }
    maxs=max(maxs,m-son[p]);
    if(maxs<mins){
        root=p;
        mins=maxs;
    }
}

void DFS2(int anc,int pre,int p,int w){
    son[p]=1;
    f[anc][w%3]++;
    for(int i=0;i<G[p].size();i++){
        Data q=G[p][i];
        if(q.v==pre||del[q.v]) continue;
        DFS2(anc,p,q.v,(w+q.w)%3);
        son[p]+=son[q.v];
    }
}

void work(int p,int tot){
    int r;
    int sum[3]={0};

    m=tot;mins=INF;
    DFS1(0,p);
    r=root;
    for(int i=0;i<G[r].size();i++){
        Data q=G[r][i];
        if(del[q.v]) continue;
        f[q.v][0]=f[q.v][1]=f[q.v][2]=0;
        DFS2(q.v,r,q.v,q.w%3);
        for(int i=0;i<=2;i++) sum[i]+=f[q.v][i];
    }
    for(int i=0;i<G[r].size();i++){
        Data q=G[r][i];
        if(del[q.v]) continue;
        a+=f[q.v][0]*(sum[0]-f[q.v][0]);
        a+=f[q.v][1]*(sum[2]-f[q.v][2]);
        a+=f[q.v][2]*(sum[1]-f[q.v][1]);
        a+=(f[q.v][0]<<1);
    }
    del[r]=1;
    for(int i=0;i<G[r].size();i++){
        Data q=G[r][i];
        if(del[q.v]) continue;
        work(q.v,son[q.v]);
    }
}

int gcd(int x,int y){
    return y?gcd(y,x%y):x;
}

int main(){
    read();
    work(1,n);
    int g=gcd(a,n*n);
    printf("%d/%d",a/g,n*n/g);

    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值