【问题描写叙述】
由爸爸在纸上画n个“点”。并用n-1条“边”把这n个“点”恰好连通(事实上这就是一棵树)。
而且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点(当然他们选点时是看不到这棵树的),假设两个点之间全部边上数的和加起来恰好是3的倍数,则判聪聪赢,否则可可赢。
聪聪很爱思考问题,在每次游戏后都会细致研究这棵树,希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。
【输入格式】
后面n-1行,每行3个整数x、y、w,表示x号点和y号点之间有一条边,上面的数是w。
【输出格式】
【例子输入】
1 2 1
1 3 2
1 4 1
2 5 3
【例子输出】
【例子说明】
对于100%的数据。n<=20000
#include<iostream> #include<cstring> #include<cstdio> using namespace std; int temp,x,y,z,n,cnt,f[100001],son[100001],ans,root,sum,point[1000001],next[1000001],d[100001],num[10000001]; struct use{int st,en,val;}b[10000001]; bool isfocus[100001]; void add(int x,int y,int z){ z%=3; next[++cnt]=point[x];point[x]=cnt; b[cnt].st=x;b[cnt].en=y;b[cnt].val=z; } void findfocus(int x,int fa) { son[x]=1;f[x]=0; for (int i=point[x];i;i=next[i]) if (!isfocus[b[i].en]&&b[i].en!=fa) { findfocus(b[i].en,x); son[x]+=son[b[i].en]; f[x]=max(f[x],son[b[i].en]); } f[x]=max(f[x],sum-son[x]); if (f[x]<f[root]) root=x; } void getdeep(int x,int fa) { num[d[x]]++; for (int i=point[x];i;i=next[i]) if(!isfocus[b[i].en]&&b[i].en!=fa) { d[b[i].en]=(d[x]+b[i].val)%3; getdeep(b[i].en,x); } } int cal(int x,int now) { num[0]=num[1]=num[2]=0; d[x]=now; getdeep(x,0); return 2*num[1]*num[2]+num[0]*num[0]; } void work(int x) { ans+=cal(x,0); isfocus[x]=true; for (int i=point[x];i;i=next[i]) if (!isfocus[b[i].en]) { ans-=cal(b[i].en,b[i].val); sum=son[b[i].en];root=0; findfocus(b[i].en,0); work(root); } } int gcd(int a,int b){return b==0?
a:gcd(b,a%b);} int main() { freopen("cckk.in","r",stdin); freopen("cckk.out","w",stdout); scanf("%d",&n); memset(isfocus,false,sizeof(isfocus)); for (int i=1;i<=n-1;i++) { scanf("%d%d%d",&x,&y,&z); add(x,y,z);add(y,x,z); } f[0]=sum=n; findfocus(1,0); work(root); temp=gcd(ans,n*n); cout<<ans/temp<<"/"<<n*n/temp<<endl; }