【bzoj 2152】聪明可可

最新推荐文章于 2019-07-15 22:04:01 发布

原创最新推荐文章于 2019-07-15 22:04:01 发布 · 395 阅读

0 ·

CC 4.0 BY-SA版权

bzoj 同时被 2 个专栏收录

60 篇文章

订阅专栏

点分治

4 篇文章

订阅专栏

本文介绍了一种使用点分治解决特定类型问题的方法，并通过一个示例代码详细展示了该算法的具体实现过程。文章中涉及了如何构建图结构、深度优先搜索、子树大小计算等关键步骤。

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

传送门~

解题思路

点分治。
代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<string>
#include<cmath>
#include<cstdlib>
using namespace std;
int hed[20005],nex[40005],lb[40005],cap[40005];
int f[5],g[5];
bool pa[20005];
int n,lo,u,ans,siz;
void add(int x,int y,int num){
    lo++;
    nex[lo]=hed[x];
    hed[x]=lo;
    lb[lo]=y;
    cap[lo]=num;
}
void dfs(int x,int fa,int num){
    f[num]++;
    for(int i=hed[x];i!=0;i=nex[i])
    if(lb[i]!=fa && !pa[lb[i]])
    dfs(lb[i],x,(num+cap[i])%3);
}
int getsiz(int x,int fa){
    int rc=0;
    for(int i=hed[x];i!=0;i=nex[i])
    if(!pa[lb[i]] && lb[i]!=fa)
    rc+=getsiz(lb[i],x);
    return rc+1;
}
int getu(int x,int fa){
    int rc=0,flag=0;
    for(int i=hed[x];i!=0;i=nex[i])
    if(!pa[lb[i]] && lb[i]!=fa){
        int r0=getu(lb[i],x);
        if(r0*2>siz) flag=1;
        rc+=r0; 
    }
    if((rc+1)*2<siz) flag=1;
    if(!flag) u=x;
    return rc+1;
}
void solve(int x){
    siz=getsiz(x,x);
    int fzbl=getu(x,x);
    pa[u]=1;g[0]=1;
    for(int i=hed[u];i!=0;i=nex[i])
    if(!pa[lb[i]]){
        dfs(lb[i],u,cap[i]);
        ans+=f[0]*g[0]+f[1]*g[2]+f[2]*g[1];
        g[0]+=f[0];g[1]+=f[1];g[2]+=f[2];
        f[0]=f[1]=f[2]=0;
    }
    g[0]=g[1]=g[2]=0;
    for(int i=hed[u];i!=0;i=nex[i])
    if(!pa[lb[i]])
    solve(lb[i]);
}
int main(){
    int ai,bi,ci;
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&ai,&bi,&ci);ci%=3;
        add(ai,bi,ci);add(bi,ai,ci);
    }
    solve(1);
    ans=ans*2+n;
    ai=__gcd(ans,n*n);
    printf("%d/%d",ans/ai,n*n/ai);
    return 0;
}