BZOJ 1989 概率相关

思路:
一条边免费的概率为 (经过它的路/总路径条数)^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;
    }
}

这里写图片描述

posted @ 2017-01-29 22:07  SiriusRen  阅读(96)  评论(0编辑  收藏  举报