聪聪和可可是兄弟俩,他们俩经常为了一些琐事打起来,例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑(可是他们家只有一台电脑)……遇到这种问题,一般情况下石头剪刀布就好了,可是他们已经玩儿腻了这种低智商的游戏。
他们的爸爸快被他们的争吵烦死了,所以他发明了一个新游戏:由爸爸在纸上画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;
}