专栏首页饶文津的专栏【UVALive - 6534 】Join two kingdoms (树的直径的期望)

【UVALive - 6534 】Join two kingdoms (树的直径的期望)

bupt spring training for 2015 #2nd J

题意

给两棵树,分别有 n,m 个节点(1 ≤ N, Q ≤ 4 × 10^4),等概率连接属于不同树的两个节点,求新树的直径(最远两点的距离)的期望。

题解

代码

#include<bits/stdc++.h>
#define ll long long
#define N 100005
using namespace std;
struct edge{int to,next;}e[N<<1];
int head[N],cnt;
void add(int u,int v){e[cnt]=(edge){v,head[u]};head[u]=cnt++;}
void init(){memset(head,-1,sizeof head);cnt=0;}
int d[N],d2[N],ds[N];
int deep,dex;
ll s[N],ans;
void dfs(int x,int fa){
    if(d[x]>=deep)deep=d[x],dex=x;
    for(int i=head[x];~i;i=e[i].next){
        int v=e[i].to;
        if(v==fa)continue;
        d[v]=d[x]+1;
        dfs(v,x);
    }
}
int getDep(int x,int ed){
    deep=0;d[x]=0;dfs(x,0);
    d[dex]=0;dfs(dex,0);
    for(int i=x;i<=ed;++i)d2[i]=d[i];
    d[dex]=0;dfs(dex,0);
    for(int i=x;i<=ed;++i)ds[i]=max(d[i],d2[i]);
    return deep;
}
int main(){
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        init();ans=0;
        for(int u,v,i=1;i<n;++i)scanf("%d%d",&u,&v),add(u,v),add(v,u);
        for(int u,v,i=1;i<m;++i)scanf("%d%d",&u,&v),add(u+n,v+n),add(v+n,u+n);

        int len=max(getDep(1,n),getDep(n+1,n+m));

        sort(ds+1,ds+1+n);
        for(int i=1;i<=n;++i)s[i]=s[i-1]+ds[i];
        for(int i=1;i<=m;++i){
            int b=lower_bound(ds+1,ds+1+n,len-ds[i+n]-1)-ds-1;
            ans+=(ll)b*len+(ll)(n-b)*ds[i+n]+(n-b)+s[n]-s[b];
        }
        printf("%.3f\n",ans*1./(n*m));
    }
    return 0;
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 【CodeForces 730H】Car Repair Shop

    模拟,先看从s[i]时刻开始修理,和之前i-1个是否冲突。如果冲突,就枚举每个s[j]+d[j]时刻开始,看是否冲突,再从中选择最小的时刻。

    饶文津
  • 【POJ 3321】Apple Tree

    有n个节点以1为根节点的树,给你树的边关系u-v,一开始每个节点都有一个苹果,接下来有两种操作,C x改变节点x的苹果状态,Q x查询x为根的树的所有苹果个数。

    饶文津
  • 【CCCC天梯赛决赛】

    cccc的天梯赛决赛,水题一样的水,中档题以上的还是没做出来。补了一下题,觉得其实也不是很难,主要是练的少。

    饶文津
  • 用邻接链表存图 详讲

    在许多图论的题目中,我们首先要存图,之前我已经学习了用邻接矩阵存图 ,但是看许多大佬都是用邻接表存图,觉得还是学习下好!

    用户7727433
  • 算法篇:求1的个数

    https://leetcode-cn.com/problems/number-of-1-bits/

    灰子学技术
  • 【CodeForces 730H】Car Repair Shop

    模拟,先看从s[i]时刻开始修理,和之前i-1个是否冲突。如果冲突,就枚举每个s[j]+d[j]时刻开始,看是否冲突,再从中选择最小的时刻。

    饶文津
  • 2015-偶数求和

    有一个长度为n(n<=100)的数列,该数列定义为从2开始的递增有序偶数,现在要求你按照顺序每m个数求出一个平均值,如果最后不足m个,则以实际数量求平均值。编程...

    用户2038589
  • Java实现图片的滤镜效果滤镜实现总结

    在移动端或者在web开发时处理图片都是一件麻烦的事儿。我调研过很多library,特别是在移动端处理图片时动不动都需要使用 C++ 或者 OpenCV。这对于 ...

    fengzhizi715
  • 数组的简单练习题

    1.将一个给定的整型数组转置输出, 例如: 源数组,1 2 3 4 5 6 转置之后的数组,6 5 4 3 2 1

    阮键
  • 洛谷P1437 [HNOI2004]敲砖块(dp)

    在一个凹槽中放置了 n 层砖块、最上面的一层有n 块砖,从上到下每层依次减少一块砖。每块砖

    attack

扫码关注云+社区

领取腾讯云代金券