聪聪可可

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值