国家集训队 聪聪可可 题解

题目传送门

题目大意： 给出一棵树，求树上有多少条路径的权可以被 $3$ 整除以及总路径数。

题解

显然，总路径数很容易就可以用 $O(1)$ 的时间求出来，问题就在于，如何求树上有多少条路径的权可以被 $3$ 整除，那么，这种树上路径统计问题，肯定是要用树上点分治嘛！

代码如下：

#include <cstdio>
#include <cstring>
#define maxn 20010
#define inf 999999999

int n,len;
struct node{int x,y,z,next;};
node e[40010];
int first[maxn];
int ms,root,Size;
int ans1,ans2;
void buildroad(int x,int y,int z)
{
    len++;
    e[len].x=x;
    e[len].y=y;
    e[len].z=z;
    e[len].next=first[x];
    first[x]=len;
}
int size[maxn],mson[maxn];
bool v[maxn];
void getroot(int x,int fa)
{
    size[x]=1;mson[x]=0;
    for(int i=first[x];i;i=e[i].next)
    {
        int y=e[i].y;
        if(y==fa||v[y])continue;
        getroot(y,x);
        size[x]+=size[y];
        if(size[y]>mson[x])mson[x]=size[y];
    }
    if(Size-size[x]>mson[x])mson[x]=Size-size[x];
    if(mson[x]<ms)ms=mson[x],root=x;
}
int dis[maxn],t;
void getdis(int x,int fa,int z)
{
    dis[++t]=z;
    for(int i=first[x];i;i=e[i].next)
    {
        int y=e[i].y;
        if(y==fa||v[y])continue;
        getdis(y,x,z+e[i].z);
    }
}
int sum[5];//sum[i]表示模3余i的dis的总数量
void solve(int x,int y,int id)
{
    t=0;
    getdis(x,0,y);
    memset(sum,0,sizeof(sum));
    for(int i=1;i<=t;i++)
    sum[dis[i]%3]++;
    ans1+=(sum[1]*sum[2]+(sum[0]-1)*sum[0]/2)*id;
}
void fenzhi(int x,int ssize)
{
    v[x]=true;
    solve(x,0,1);
    for(int i=first[x];i;i=e[i].next)
    {
        int y=e[i].y;
        if(v[y])continue;
        solve(y,e[i].z,-1);
        ms=inf;root=0;Size=size[y]<size[x]?size[y]:(ssize-size[x]);
        getroot(y,0);
        fenzhi(root,(size[y]<size[x]?size[y]:(ssize-size[x])));
    }
}
int gcd(int x,int y)
{
    if(y==0)return x;
    else return gcd(y,x%y);
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y,z;
        scanf("%d %d %d",&x,&y,&z);
        buildroad(x,y,z);
        buildroad(y,x,z);
    }
    ms=inf,root=0,Size=n;
    getroot(1,0);
    fenzhi(root,n);
    ans1*=2;ans1+=n;//*2是因为(x,y)和(y,x)算两种情况，+n是要算上(x,x)的情况
    ans2=n*n;
    int lalala=gcd(ans1,ans2);
    printf("%d/%d",ans1/lalala,ans2/lalala);
}