bzoj2152 聪聪可可

求树中长度是3的倍数的路径个数。

关键词：路径统计、点分治法。

关键在于判断如何判断路径是否满足条件和如何计算路径条数。本题用t[i]数组记录i到达根节点的距离为i的条数，t[1]*t[2]*2+t[0]*t[0]即为所求。//为什么是t[0]*t[0]？？？

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#define maxn 20010
#define lowbit(x) x&(-x)
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;

int n;
struct node{
    int to,next,w;
}edge[maxn*2];
int head[maxn],tot;
int vis[maxn];
int f[maxn],son[maxn],core,sum;
int d[maxn],t[5];
int ans;

void init(){
    mem(head,-1);
    tot=ans=0;
    mem(vis,0);
}

void add(int a,int b,int c){
    edge[tot].to=b,edge[tot].next=head[a],edge[tot].w=c;
    head[a]=tot++;
}

void getcore(int u,int fa){
    son[u]=1,f[u]=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v]||v==fa) continue;
        getcore(v,u);
        son[u]+=son[v];
        f[u]=max(f[u],son[v]);
    }
    f[u]=max(f[u],sum-son[u]);
    if(f[u]<f[core]) core=u;//f[u]算出后取最小的一个
}

void getdeep(int u,int fa){//计算u所在子树每个点相对u的深度+u自身的深度
    t[d[u]]++;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to,w=edge[i].w;
        if(vis[v]||v==fa) continue;
        d[v]=(d[u]+w)%3;
        getdeep(v,u);
    }
}

int cal(int u,int w){
    d[u]=w%3;
    t[0]=t[1]=t[2]=0;//初始化！
    getdeep(u,0);
    printf("%d %d %d %d\n",u,t[0],t[1],t[2]);
    return t[1]*t[2]*2+t[0]*t[0];
}

void work(int u){//点分治主过程
    ans+=cal(u,0);//求经过重心u的合法路径数
    vis[u]=1;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to,w=edge[i].w;
        if(vis[v]) continue;
        ans-=cal(v,w);//减去仅属于一个分支的合法路径数（不符合路径定义）
        sum=son[v],core=0;
        getcore(v,0);
        work(core);
    }
}

int gcd(int a,int b){
    return b==0?a:gcd(b,a%b);
}

int main(){
    freopen("a.txt","r",stdin);
    scanf("%d",&n);
    init();
    for(int i=0;i<n-1;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c%3),add(b,a,c%3);
    }
    core=0,sum=n,f[0]=INF;
    getcore(1,0);
    work(core);
    int tmp=gcd(ans,n*n);
    printf("%d/%d\n",ans/tmp,n*n/tmp);
    return 0;
}