Description
聪聪和可可是兄弟俩,他们俩经常为了一些琐事打起来,例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑(可是他们家只有一台电脑)……遇到这种问题,一般情况下石头剪刀布就好了,可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了,所以他发明了一个新游戏:
由爸爸在纸上画n个“点”,并用n-1条“边”把这n个“点”恰好连通(其实这就是一棵树)。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点(当然他们选点时是看不到这棵树的),如果两个点之间所有边上数的和加起来恰好是3的倍数,则判聪聪赢,否则可可赢。
聪聪非常爱思考问题,在每次游戏后都会仔细研究这棵树,希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。
Input
输入的第1行包含1个正整数n。
后面n-1行,每行3个整数x、y、w,表示x号点和y号点之间有一条边,上面的数是w。
Output
以即约分数形式输出这个概率(即“a/b”的形式,其中a和b必须互质。如果概率为1,输出“1/1”)。
Sample Input
5
1 2 1
1 3 2
1 4 1
2 5 3
Sample Output
13/25
Hint
【样例说明】
13组点对分别是(1,1) (2,2) (2,3) (2,5) (3,2) (3,3) (3,4) (3,5) (4,3) (4,4) (5,2) (5,3) (5,5)。
【数据规模】
对于30%的数据,n<=1000;
另有20%的数据,给出的树中每个节点的度不超过2;
聪聪和可可是兄弟俩,他们俩经常为了一些琐事打起来,例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑(可是他们家只有一台电脑)……遇到这种问题,一般情况下石头剪刀布就好了,可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了,所以他发明了一个新游戏:
由爸爸在纸上画n个“点”,并用n-1条“边”把这n个“点”恰好连通(其实这就是一棵树)。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点(当然他们选点时是看不到这棵树的),如果两个点之间所有边上数的和加起来恰好是3的倍数,则判聪聪赢,否则可可赢。
聪聪非常爱思考问题,在每次游戏后都会仔细研究这棵树,希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。
Input
输入的第1行包含1个正整数n。
后面n-1行,每行3个整数x、y、w,表示x号点和y号点之间有一条边,上面的数是w。
Output
以即约分数形式输出这个概率(即“a/b”的形式,其中a和b必须互质。如果概率为1,输出“1/1”)。
Sample Input
5
1 2 1
1 3 2
1 4 1
2 5 3
Sample Output
13/25
Hint
【样例说明】
13组点对分别是(1,1) (2,2) (2,3) (2,5) (3,2) (3,3) (3,4) (3,5) (4,3) (4,4) (5,2) (5,3) (5,5)。
【数据规模】
对于30%的数据,n<=1000;
另有20%的数据,给出的树中每个节点的度不超过2;
对于100%的数据,n<=20000。
最简单的点分治,建边时把边权mod 3,加值的时候也mod 3,就可以解决了
下面是注释和代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define size 400005
using namespace std;
struct Edge
{
int to,next,v;
}w[size*4];
int h[size]={0},cnt=0;
int n,d[size]={0};
bool vis[size]={0};
int hash[5]={0};
int Size[size]={0};
int gcd(int a,int b)
{
if(b==0)return a;
else return gcd(b,a%b);
}
void add(int x,int y,int v)
{
cnt++;w[cnt].to=y;w[cnt].next=h[x];h[x]=cnt;w[cnt].v=v;
}
void init()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c%3);
add(b,a,c%3);
}
}
void DFS2(int x,int fa)
{
Size[x]=1;
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
if(!vis[to]&&to!=fa)
{
DFS2(to,x);
Size[x]+=Size[to];
}
}
}
int Findcenter(int x)
{
DFS2(x,0);
int c=x,tot=Size[x];
while(1)
{
int cc=0;
for(int i=h[c];i;i=w[i].next)
{
int to=w[i].to;
if(!vis[to]&&Size[to]>Size[cc])cc=to;
}
if(Size[cc]<=tot/2)break;
else
{
Size[c]-=Size[cc];
Size[cc]=tot;
c=cc;
}
}
return c;
}
void Getdep(int x,int fa)
{
hash[d[x]]++;
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
if(!vis[to]&&to!=fa)
{
d[to]=(d[x]+w[i].v)%3;
Getdep(to,x);
}
}
}
int Calc(int x,int len)
{
hash[0]=hash[1]=hash[2]=0;
d[x]=len%3;
Getdep(x,0);
return hash[0]*hash[0]+hash[1]*hash[2]*2;
/*
一共选两边,<span style="font-family: Arial, Helvetica, sans-serif;">边权已经每步mod 3,现在要得到3(取余为0),也就是被3整除,只能为0+0或者1+2或2+1,乘法原理+加法原理</span>
<span style="font-family: Arial, Helvetica, sans-serif;">*/</span>
}
int ans=0;
void DFS(int x)
{
int G=Findcenter(x);
vis[G]=1;
ans+=Calc(G,0);
for(int i=h[G];i;i=w[i].next)
{
int to=w[i].to;
if(!vis[to])ans-=Calc(to,w[i].v),DFS(to);
}
}
int main(){
init();
DFS(1);
int GCD=gcd(ans,n*n);
printf("%d/%d\n",ans/GCD,n*n/GCD);
return 0;
}
代码来自hzwer