2152: 聪聪可可
Time Limit: 3 Sec Memory Limit: 259 MB
Submit: 2065 Solved: 1081
[Submit][Status][Discuss]
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
【样例说明】
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)。
【数据规模】
对于100%的数据,n<=20000。
HINT
Source
[Submit][Status][Discuss]
HOME Back
分析:
聪聪可可这两个人,和佳佳,淘淘在OI中都是很迷得存在
对于路径,还是两种情况:经过根结点,不经过根结点
但是这次我们需要记录的是路径和为3的倍数的路径
像朴素的点分治一样,我们还是找到树的重心,进行dfs
每一条路径被根结点分成两段(下文就称两条链为x1,x2),这道题要求的是路径和为3的倍数的路径
实际上我们可以直接列出这些情况:
- sum[v1]%3==0,sum[v2]%3==0
- sum[v1]%3==1,sum[v2]%3==2
- sum[v1]%3==2,sum[v2]%3==1
所以我们在计算的时候,直接统计权值和%3=0,%3=1,%3=2的树链条数
这样直接计算即可
int cal(int now)
{
for (int i=0;i<3;i++) d[i]=0;
getdeep(now,0);
int t=0;
t+=d[0]*d[0];
t+=d[1]*d[2];
t+=d[2]*d[1];
return t;
}
最后的答案就是ans/n*n
为什么这里的路径条数是n*n呢
因为ans统计的时候,我们就没有按照有序点对计算
//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N=20010;
int deep[N],sz[N],vis[N],d[3],f[N];
struct node{
int x,y,v,nxt;
};
node way[N<<1];
int st[N],tot=0,n,m,sum,root,ans;
void add(int u,int w,int z)
{
tot++;
way[tot].x=u;way[tot].y=w;way[tot].v=z;way[tot].nxt=st[u];st[u]=tot;
}
void getroot(int now,int fa) //朴素的树的重心
{
f[now]=0;
sz[now]=1;
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa&&vis[way[i].y]==0)
{
getroot(way[i].y,now);
sz[now]+=sz[way[i].y];
f[now]=max(f[now],sz[way[i].y]);
}
f[now]=max(f[now],sum-sz[now]);
if (f[now]<f[root]) root=now;
}
void getdeep(int now,int fa)
{
d[deep[now]%3]++; //统计树链条数
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa&&vis[way[i].y]==0)
{
deep[way[i].y]=deep[now]+way[i].v;
deep[way[i].y]%=3;
getdeep(way[i].y,now);
}
}
int cal(int now)
{
for (int i=0;i<3;i++) d[i]=0;
getdeep(now,0);
int t=0;
t+=d[0]*d[0];
t+=d[1]*d[2];
t+=d[2]*d[1];
return t;
}
void dfs(int now)
{
deep[now]=0; //子树内的deep
vis[now]=1;
ans+=cal(now);
for (int i=st[now];i;i=way[i].nxt)
if (vis[way[i].y]==0)
{
deep[way[i].y]=way[i].v;
ans-=cal(way[i].y); //去重
sum=sz[way[i].y];
root=0;
getroot(way[i].y,0);
dfs(root);
}
}
int gcd(int a,int b)
{
int r=a%b;
while (r)
{
a=b;
b=r;
r=a%b;
}
return b;
}
int main()
{
scanf("%d",&n);
for (int i=1;i<n;i++)
{
int u,w,z;
scanf("%d%d%d",&u,&w,&z);
add(u,w,z%3);
add(w,u,z%3);
}
f[0]=N;sum=n;
root=0;
ans=0;
getroot(1,0);
dfs(root);
if (ans==0)
{
printf("0/0");
return 0;
}
int t=n*n; //总方案数
int t1=gcd(t,ans);
printf("%d/%d\n",ans/t1,t/t1);
return 0;
}