BZOJ 1989 概率相关

最新推荐文章于 2022-02-10 16:17:46 发布

Sirius_Ren

最新推荐文章于 2022-02-10 16:17:46 发布

阅读量500

点赞数

CC 4.0 BY-SA版权

分类专栏：搜索 DP 文章标签： dfs

本文链接：https://blog.youkuaiyun.com/qq_31785871/article/details/54774178

DP 同时被 2 个专栏收录

84 篇文章

订阅专栏

搜索

78 篇文章

订阅专栏

本文介绍了一种计算图中边被免费选择概率的算法。通过计算每条边被选中的概率等于该边所连接子图大小乘以其补集的平方，使用深度优先搜索(DFS)遍历整个图来实现。注意在计算过程中避免整数溢出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：
一条边免费的概率为（经过它的路/总路径条数）^2
DFS即可

有个地方没有用 long long炸了好久…

//By SiriusRen
#include <cstdio>
using namespace std;
const int N=20050;
int n,m,xx,yy,first[N],next[N],v[N],tot,size[N];
typedef long long ll;ll ans;
void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
void dfs(int x,int fa){
    size[x]=1;
    for(int i=first[x];i;i=next[i])if(v[i]!=fa)
        dfs(v[i],x),size[x]+=size[v[i]];
    for(int i=first[x];i;i=next[i])if(v[i]!=fa)
        ans+=(ll(n-size[v[i]])*size[v[i]])*((n-size[v[i]])*size[v[i]]);
}
int main(){
    while(~scanf("%d",&n)){
        tot=1;ans=0;m=n*(n-1)/2;
        for(int i=1;i<n;i++)scanf("%d%d",&xx,&yy),add(xx,yy),add(yy,xx);
        dfs(1,-1);
        printf("%.6lf\n",(double)ans/m/m);
        for(int i=1;i<=n;i++)first[i]=0;
    }
}