BZOJ 2152 聪聪可可 [树分治 or 树形DP]

Description

聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：由爸爸在纸上画n个“点”，并用n-1条“边”把这n个“点”恰好连通（其实这就是一棵树）。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点（当然他们选点时是看不到这棵树的），如果两个点之间所有边上数的和加起来恰好是3的倍数，则判聪聪赢，否则可可赢。聪聪非常爱思考问题，在每次游戏后都会仔细研究这棵树，希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。

题意：给出一棵树，边有权，问随机取两点，其路径上的边权和整除3的概率，以最简分数的形式给出答案。

解法：

1)因为取模数很小，可以考虑树形DP，记Cot[X][3]为点X的子树中，距离根节点距离%3分别为0,1,2的点数，那么就可以很简单的统计出经过X点的路径数，累加便是答案。（计算方法见代码），O（N）复杂度，150+MS

2)如果取模数很大，可以用树分治来写（例题  HDU 4812），计算经过分治重心的路径数，方法类似于树形DP，取模数非常大的时候，需要另外进行hash（因为N远小于MOD，所以hash复杂度可以看做O（1）），整体复杂度为O（NlogN） 此题300+MS

树形DP代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<stdlib.h>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<bitset>
#pragma comment(linker, "/STACK:1024000000,1024000000")
template <class T>
bool scanff(T &ret){ //Faster Input
    char c; int sgn; T bit=0.1;
    if(c=getchar(),c==EOF) return 0;
    while(c!='-'&&c!='.'&&(c<'0'||c>'9')) c=getchar();
    sgn=(c=='-')?-1:1;
    ret=(c=='-')?0:(c-'0');
    while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
    if(c==' '||c=='\n'){ ret*=sgn; return 1; }
    while(c=getchar(),c>='0'&&c<='9') ret+=(c-'0')*bit,bit/=10;
    ret*=sgn;
    return 1;
}
#define inf 1073741823
#define llinf 4611686018427387903LL
#define PI acos(-1.0)
#define lth (th<<1)
#define rth (th<<1|1)
#define rep(i,a,b) for(int i=int(a);i<=int(b);i++)
#define drep(i,a,b) for(int i=int(a);i>=int(b);i--)
#define gson(i,root) for(int i=ptx[root];~i;i=ed[i].next)
#define tdata int testnum;scanff(testnum);for(int cas=1;cas<=testnum;cas++)
#define mem(x,val) memset(x,val,sizeof(x))
#define mkp(a,b) make_pair(a,b)
#define findx(x) lower_bound(b+1,b+1+bn,x)-b
#define pb(x) push_back(x)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

#define NN 100100
int ptx[NN],lnum;
struct edge{
    int v,next,w;
    edge(){}
    edge(int v,int next,int w){
        this->v=v;
        this->next=next;
        this->w=w;
    }
}ed[NN*2];
void addline(int x,int y,int w){
    ed[lnum]=edge(y,ptx[x],w);
    ptx[x]=lnum++;
}
int n,ans;
int dis[NN];
int c[NN][3];
void dfs(int x,int fa,int d){
    c[x][d%3]++;
    ans++;
    gson(i,x){
        int y=ed[i].v;
        if(y==fa)continue;
        dfs(y,x,d+ed[i].w);
        if(d%3==0){
            ans+=c[x][0]*c[y][0]*2;
            ans+=c[x][1]*c[y][2]*2;
            ans+=c[x][2]*c[y][1]*2;
        }
        if(d%3==1){
            ans+=c[x][2]*c[y][0]*2;
            ans+=c[x][0]*c[y][2]*2;
            ans+=c[x][1]*c[y][1]*2;
        }
        if(d%3==2){
            ans+=c[x][1]*c[y][0]*2;
            ans+=c[x][0]*c[y][1]*2;
            ans+=c[x][2]*c[y][2]*2;

        }
        c[x][0]+=c[y][0];
        c[x][1]+=c[y][1];
        c[x][2]+=c[y][2];

    }

}
int main(){
    mem(ptx,-1);
    scanff(n);
    rep(i,2,n){
        int x,y,z;
        scanff(x);
        scanff(y);
        scanff(z);
        addline(x,y,z);
        addline(y,x,z);
    }
    dfs(1,0,0);
    int g=__gcd(ans,n*n);
    printf("%d/%d\n",ans/g,n*n/g);
    return 0;
}

树分治代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<stdlib.h>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<bitset>
#pragma comment(linker, "/STACK:1024000000,1024000000")
template <class T>
bool scanff(T &ret){ //Faster Input
    char c; int sgn; T bit=0.1;
    if(c=getchar(),c==EOF) return 0;
    while(c!='-'&&c!='.'&&(c<'0'||c>'9')) c=getchar();
    sgn=(c=='-')?-1:1;
    ret=(c=='-')?0:(c-'0');
    while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
    if(c==' '||c=='\n'){ ret*=sgn; return 1; }
    while(c=getchar(),c>='0'&&c<='9') ret+=(c-'0')*bit,bit/=10;
    ret*=sgn;
    return 1;
}
#define inf 1073741823
#define llinf 4611686018427387903LL
#define PI acos(-1.0)
#define lth (th<<1)
#define rth (th<<1|1)
#define rep(i,a,b) for(int i=int(a);i<=int(b);i++)
#define drep(i,a,b) for(int i=int(a);i>=int(b);i--)
#define gson(i,root) for(int i=ptx[root];~i;i=ed[i].next)
#define tdata int testnum;scanff(testnum);for(int cas=1;cas<=testnum;cas++)
#define mem(x,val) memset(x,val,sizeof(x))
#define mkp(a,b) make_pair(a,b)
#define findx(x) lower_bound(b+1,b+1+bn,x)-b
#define pb(x) push_back(x)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

#define NN 20020
int n,ptx[NN],lnum;
int maxv[NN],maxval,sz[NN];
int dis[NN],dn;
int center;
bool vis[NN];
struct edge{
    int v,next,w;
    edge(){}
    edge(int v,int next,int w){
        this->v=v;
        this->next=next;
        this->w=w;
    }
}ed[NN*2];
void addline(int x,int y,int w){
    ed[lnum]=edge(y,ptx[x],w);
    ptx[x]=lnum++;
}

int getsize(int x,int fa){
    sz[x]=1;
    gson(i,x){
        int y=ed[i].v;
        if(y==fa||vis[y])continue;
        sz[x]+=getsize(y,x);
    }
    return sz[x];
}
void getcenter(int r,int x,int fa){
    maxv[x]=sz[r]-sz[x];
    gson(i,x){
        int y=ed[i].v;
        if(y==fa||vis[y])continue;
        getcenter(r,y,x);
        maxv[x]=max(maxv[x],sz[y]);
    }
    if(maxv[x]<maxval)maxval=maxv[x],center=x;
}
void getdis(int x,int fa,int d){
    dis[++dn]=d;
    gson(i,x){
        int y=ed[i].v;
        if(y==fa||vis[y])continue;
        getdis(y,x,d+ed[i].w);
    }
}
int c[3];
int calc(int x,int d){
    dn=c[1]=c[2]=0;
    c[0]=1;
    int sum=1;
    gson(i,x){
        int y=ed[i].v;
        if(vis[y])continue;
        dn=0;
        getdis(y,x,ed[i].w);
        rep(j,1,dn){
            int k=dis[j]%3;
            if(k==0)sum+=c[0]*2;
            else if(k==1)sum+=c[2]*2;
            else if(k==2)sum+=c[1]*2;
        }
        rep(j,1,dn)c[dis[j]%3]++;
    }
    return sum;
}
int ans;
void solve(int x){
    maxval=inf;
    getsize(x,0);
    getcenter(x,x,0);
    x=center;
    vis[x]=1;
    ans+=calc(x,0);
    gson(i,x){
        int y=ed[i].v;
        if(vis[y])continue;
        //如果ans-=calc()要放在solve()之前，否则vis会改变
        solve(y);
    }
}

int main(){
    scanff(n);
    mem(ptx,-1);
    rep(i,1,n-1){
        int x,y,w;
        scanff(x);scanff(y);scanff(w);
        addline(x,y,w);addline(y,x,w);
    }
    solve(1);
    int x=ans;
    int y=n*n;
    int g=__gcd(x,y);
    x/=g;
    y/=g;
    printf("%d/%d\n",x,y);
}
/*
3
1 2 1
1 3 2


*/