聪聪可可 洛谷2634 bzoj 2152

题目

聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。

他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：由爸爸在纸上画n个“点”，并用n-1条“边”把这n个“点”恰好连通（其实这就是一棵树）。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点（当然他们选点时是看不到这棵树的），如果两个点之间所有边上数的和加起来恰好是3的倍数，则判聪聪赢，否则可可赢。

聪聪非常爱思考问题，在每次游戏后都会仔细研究这棵树，希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。

分析

很简单的点分治。
利用两个数组g[]表示搜当前根的子树时，当前子树之前的路径长ｘ的方案数，ff表示当前子树路径长ｘ方案数
ans+=g[j]*ff[(3-j)%3]*2;
//注意(1,1)合法,(1,2),(2,1)算两种

ps:我tm之前的找重心写错了。。。

code

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<string>
#include<cmath>

using namespace std;

const int maxn=20000+10;

struct arr{
    int x,y;
    int w;
    int next;
    int flag;
}edge[maxn*2];
int ls[maxn];
int edge_m;

long long ans;

int n,k;

bool done[maxn];

void add(int x,int y,int w)
{
    edge[++edge_m]=(arr){x,y,w,ls[x]},ls[x]=edge_m;
    edge[++edge_m]=(arr){y,x,w,ls[y]},ls[y]=edge_m;
}

int sz[maxn];
int f[maxn];
int rt,size;

void getrt(int x,int fa)
{
    sz[x]=1;
    f[x]=0;
    for(int i=ls[x];i;i=edge[i].next)
    {
        int u=edge[i].y;
        if(u==fa||done[u]) continue;
        getrt(u,x);
        sz[x]+=sz[u];
        f[x]=max(f[x],sz[u]);
    }
    f[x]=max(f[x],size-sz[x]);
    if(f[x]<f[rt]) rt=x;
}

int dis[maxn];  
long long bucket[5];
long long ff[5];
int tot=0;

void dfs(int x,int r)
{
    sz[x]=1;
    int xx=tot;
    ff[dis[tot]]++;
    for (int i=ls[x];i;i=edge[i].next)
    {
        if (done[edge[i].y]) continue;
        if (edge[i].y==r) continue;
        dis[++tot]=0;
        dis[tot]=(dis[xx]+edge[i].w)%3;
        dfs(edge[i].y,x);
        sz[x]+=sz[edge[i].y];
    }
}

void cale(int x)
{
    memset(bucket,0,sizeof(bucket));
    bucket[0]=1;
    tot=1;
    dis[tot]=0;
    for (int i=ls[x];i;i=edge[i].next)
    {
        if (done[edge[i].y]) continue;
        int xx=tot+1;
        dis[++tot]=0;
        dis[tot]=edge[i].w%3;
        dfs(edge[i].y,x);
        for (int j=0;j<3;j++)
            ans+=bucket[j]*ff[(3-j)%3];
        for (int j=0;j<3;j++)
        {
            bucket[j]+=ff[j];
            ff[j]=0;
        }
    }
}

int work(int x)
{
    done[x]=1;
    cale(x);
    for (int i=ls[x];i;i=edge[i].next)
    {
        if (done[edge[i].y]) continue;
        f[0]=size=sz[edge[i].y];
        getrt(edge[i].y,rt=0);
        work(rt);
    }
}

int m;

long long gcd(long long a,long long b)
{
    if(b==0) return a;
    return gcd(b,a%b);
}

int main()
{
    scanf("%d",&n);
    edge_m=0;
    for (int i=1;i<n;i++)
    {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);
    }
    work(1);
    ans*=2; ans+=n;
    long long nn=n*n;
    long long d=gcd(ans,nn);
    ans/=d; nn/=d;
    printf("%lld",ans);
    printf("/");
    printf("%lld",nn);
}