和Leo一起做爱树上结构的好孩子之点分治:聪聪可可

版权声明:LeoJAM Presents https://blog.csdn.net/fcb_x/article/details/82807651

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

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

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

树上点分治:

维护对3的余数

然后就很板了

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
using namespace std;
const int N=2e5+100;
inline int GCD(int x,int y){
    while(y){
        int tmp=x%y;
        x=y;
        y=tmp;
    }
    return x;
}
struct Front_star{
    int u,v,w,nxt;
}e[N<<2];
int ecnt=0;
int first[N];
void add(int u,int v,int w){
    ++ecnt;
    e[ecnt].u=u;
    e[ecnt].v=v;
    e[ecnt].w=w;
    e[ecnt].nxt=first[u];
    first[u]=ecnt;
}
//
int ans=0;
int n;
int All;
int root;
int F[N]; 
int siz[N];
int vis[N];
void Get_Root(int u,int fat){
    siz[u]=1;
    int now=0;
    for(int i=first[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(vis[v]||v==fat)continue;
        Get_Root(v,u);
        siz[u]+=siz[v];
        if(now<siz[v])now=siz[v];
    }
    if(now<All-siz[u])now=All-siz[u];
    F[u]=now;
    if(F[root]>now)root=u;
} 
int cnt[4];
int d[N];
void Get_deep(int u,int fat){
    ++cnt[d[u]];
    for(int i=first[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(v==fat||vis[v])continue;
        d[v]=(d[u]+e[i].w)%3;
        Get_deep(v,u);
    }
}
int Solve(int u,int len){
    memset(cnt,0,sizeof(cnt));
    d[u]=len%3;
    Get_deep(u,0);
    int ret=0;
    ret+=cnt[0]*cnt[0];
    ret+=cnt[1]*cnt[2]*2;
    return ret;
}
void DFS(int u){
    ans+=Solve(u,0);
    vis[u]=1;
    int temp=siz[u];
    for(int i=first[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(vis[v])continue;
        ans-=Solve(v,e[i].w);
        All=siz[v];
//		if(All>temp) All=siz[v]-temp;
        root=0;
        Get_Root(v,0);
        DFS(root);
    }
}
int main(){
//	freopen("test.in","r",stdin);
    F[0]=1e9+7;
    scanf("%d",&n);
    for(int i=1;i<n;++i){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    root=0;
    Get_Root(1,0);
    DFS(root);
    int gcd=GCD(ans,n*n);
    cout<<ans/gcd<<"/"<<n*n/gcd;
}

猜你喜欢

转载自blog.csdn.net/fcb_x/article/details/82807651