聪聪和可可是兄弟俩,他们俩经常为了一些琐事打起来,例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑(可是他们家只有一台电脑)……遇到这种问题,一般情况下石头剪刀布就好了,可是他们已经玩儿腻了这种低智商的游戏。
他们的爸爸快被他们的争吵烦死了,所以他发明了一个新游戏:由爸爸在纸上画n个“点”,并用n-1条“边”把这n个“点”恰好连通(其实这就是一棵树)。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点(当然他们选点时是看不到这棵树的),如果两个点之间所有边上数的和加起来恰好是3的倍数,则判聪聪赢,否则可可赢。
聪聪非常爱思考问题,在每次游戏后都会仔细研究这棵树,希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。
输入格式:
输入的第1行包含1个正整数n。后面n-1行,每行3个整数x、y、w,表示x号点和y号点之间有一条边,上面的数是w。
输出格式:
以即约分数形式输出这个概率(即“a/b”的形式,其中a和b必须互质。如果概率为1,输出“1/1”)。
初步想法:点分治求出线段,然后线段相加判断下%3余数。80分。(AC的代码在后面)
改进,我们分析下上面的为什么会T2个点。
因为每次枚举了一个点算路径后,是一个n²找答案的过程。这样非常慢。。
我们思考一下,一条余数为2的线段加一个余数为1的线段。余数肯定为0.
所以我们并不需要记录是哪一条线段,只要知道余数是i的线段有几条就好了。
now表示当前子树的线段,pre表示之前子树的线段。
根据乘法原理不同子树线段直接匹配可以计算出答案。
代码实现为 一组数 不重复的两两相乘可以化简为相乘累加的形式。
例如:a*b+(a+b)c+(a+b+c)*d=ab+ac+bc+ad+bd+cd=a *(b+c+d)+b*(c+d)+cd
#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+5;
struct edge{
int to,next,w;
}e[MAXN];
struct seg{
int val,pos;
}g[MAXN<<3];
int head[MAXN],cnt=0,cnt2=0,num=0;
int tot=0,ans=0;
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}
int size[MAXN],rt,sum,dis[MAXN],f[MAXN];
bool vis[MAXN];
int n;
bool judge(int x,int y){
return(x+y)%3==0?1:0;
}
int gcd(int a,int b){
return b==0?a:gcd(b,a%b);
}
void dfs1(int u,int fa){
size[u]=1;
f[u]=0;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(vis[v]||v==fa)continue;
dfs1(v,u);
size[u]+=size[v];
f[u]=max(f[u],size[v]);
}
f[u]=max(f[u],sum-size[u]);
if(f[u]<f[rt])rt=u;
}
void dfs2(int u,int fa,int num){
size[u]=1;
g[++cnt2]=(seg){dis[u],num};
if(judge(dis[u],0))ans++;tot++;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to,w=e[i].w;
if(vis[v]||v==fa)continue;
dis[v]=dis[u]+w;
dfs2(v,u,num);
size[u]+=size[v];
}
}
void cal(int u){
cnt2=0;num=0;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to,w=e[i].w;
if(vis[v])continue;
dis[v]=w;
dfs2(v,u,++num);
}
}
void solve(int u){
vis[u]=1;
cal(u);
for(int i=1;i<cnt2;i++)
for(int j=i+1;j<=cnt2;j++)
if(g[i].pos!=g[j].pos){
tot++;
if(judge(g[i].val,g[j].val))ans++;
}
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(vis[v])continue;
sum=f[0]=size[v];
dfs1(v,rt=0);
solve(rt);
}
}
int main(){
int tem1,tem2,tem3;
scanf("%d",&n);
for(int i=1;i<n;i++){
scanf("%d%d%d",&tem1,&tem2,&tem3);
add(tem1,tem2,tem3);
add(tem2,tem1,tem3);
}
sum=f[0]=n;
rt=0;
dfs1(1,0);
solve(rt);
ans<<=1;tot<<=1;
ans+=n;tot+=n;
int tem=gcd(ans,tot);
ans/=tem;
tot/=tem;
printf("%d/%d\n",ans,tot);
}
#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+5;
struct edge{
int to,next,w;
}e[MAXN];
int head[MAXN],cnt=0,cnt2=0,num=0,now[5],pre[5];
int tot=0,ans=0;
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}
int size[MAXN],rt,sum,dis[MAXN],f[MAXN];
bool vis[MAXN];
int n;
int gcd(int a,int b){
return b==0?a:gcd(b,a%b);
}
void dfs1(int u,int fa){
size[u]=1;
f[u]=0;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(vis[v]||v==fa)continue;
dfs1(v,u);
size[u]+=size[v];
f[u]=max(f[u],size[v]);
}
f[u]=max(f[u],sum-size[u]);
if(f[u]<f[rt])rt=u;
}
void dfs2(int u,int fa,int num){
size[u]=1;
dis[u]%=3;
now[dis[u]]++;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to,w=e[i].w;
if(vis[v]||v==fa)continue;
(dis[v]=dis[u]+w)%=3;
dfs2(v,u,num);
size[u]+=size[v];
}
}
void cal(int u){
pre[0]=1;
ans++;
cnt2=0;num=0;
for(int i=head[u];i;i=e[i].next){
int v=e[i].to,w=e[i].w;
if(vis[v])continue;
(dis[v]=w)%=3;
dfs2(v,u,++num);
for(int i=0;i<3;i++){
ans+=pre[i]*now[(3-i)%3]*2;
}
for(int i=0;i<3;i++){
pre[i]+=now[i];
now[i]=0;
}
}
for(int i=0;i<3;i++)pre[i]=0;
}
void solve(int u){
vis[u]=1;
cal(u);
for(int i=head[u];i;i=e[i].next){
int v=e[i].to;
if(vis[v])continue;
sum=f[0]=size[v];
dfs1(v,rt=0);
solve(rt);
}
}
int main(){
int tem1,tem2,tem3;
scanf("%d",&n);
for(int i=1;i<n;i++){
scanf("%d%d%d",&tem1,&tem2,&tem3);
add(tem1,tem2,tem3);
add(tem2,tem1,tem3);
}
sum=f[0]=n;
rt=0;
dfs1(1,0);
solve(rt);
tot=n*n;
int tem=gcd(ans,tot);
ans/=tem;
tot/=tem;
printf("%d/%d\n",ans,tot);
}