NOI2012 : 迷失游乐园

终于补完NOI2012了好开心~

题目大意:给定一棵树或者环套外向树,求出从中随机选一条简单路径的期望长度,环上点数不超过20。

 

d[x]表示x的度数,ch[x]表示x孩子个数

up[x]表示x向上走的期望长度,down[x]表示x向下走的期望长度

f[x]表示x的父亲

 

树的情况:

 

环套外向树的情况:

先找出环,对于每棵树用之前的方法求出down[]

对环上每个点i顺时针逆时针各走一圈,求出up[i]:

up[i]=sum((i走到j的概率)*(way(i,j)+down[j])*(j往它孩子走的概率))

 

i走到j的概率分两种情况讨论,

以顺时针为例,

第一步由于要确定是顺时针还是逆时针,所以顺时针走概率为0.5,

之后每一步概率/=(上一个点孩子数+1)

 

j往它孩子走的概率分两种情况讨论,

以顺时针为例,

每一步向下走概率为该点孩子数/(该点孩子数+1),最后一步由于不可能回到i点,所以向下走概率为1

 

然后用树的方法求出其它点的up[]

再统计ans即可

 

没加任何优化居然跑了Rank1…

 

#include<cstdio>
#define N 100010
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
int n,m,i,j,x,y,z;
int g[N],nxt[N<<1],w[N<<1],v[N<<1],ed,pre[N],ch[N],d[N],fw[N],st,sum;
int cnt,a[N<<1],s[N<<1];
double up[N],down[N],ans,p;
bool vis[N],in[N];
inline void add(int x,int y,int z){d[x]++;v[++ed]=y;w[ed]=z;nxt[ed]=g[x];g[x]=ed;}
void dfs1(int x,int f){
  for(int i=g[x];i;i=nxt[i])if(v[i]!=f&&!in[v[i]])dfs1(v[i],x),ch[x]++,down[x]+=down[v[i]]+w[i];
  if(ch[x])down[x]/=ch[x];
}
void dfs2(int x,int f){
  for(int i=g[x];i;i=nxt[i])if(v[i]!=f&&!in[v[i]]){
    up[v[i]]=w[i];
    if(d[x]>1)up[v[i]]+=(up[x]*(d[x]-ch[x])+down[x]*ch[x]-w[i]-down[v[i]])/(d[x]-1);
    dfs2(v[i],x);
  }
}
void find(int x,int f,int l){
  if(st)return;
  pre[x]=f;fw[x]=l;
  if(vis[x]){st=f;return;}
  vis[x]=1;
  for(int i=g[x];i;i=nxt[i])if(v[i]!=f)find(v[i],x,w[i]);
}
int main(){
  for(read(n),read(m);i<m;i++)read(x),read(y),read(z),add(x,y,z),add(y,x,z);
  if(m<n)dfs1(1,0),dfs2(1,0);else{
    find(1,0,0);
    for(in[a[cnt=1]=st]=1,i=pre[st];i!=st;i=pre[i])in[a[++cnt]=i]=1;
    for(i=1;i<=cnt;i++)a[i+cnt]=a[i],s[i+1]=s[i+cnt+1]=fw[a[i]];
    for(i=1;i<=cnt;i++)dfs1(a[i],0);
    for(i=1;i<=cnt;i++){
      //i->j
      j=i+1;p=0.5;sum=s[j];
      up[a[i]]+=p*(sum+down[a[j]])*ch[a[j]]/(ch[a[j]]+1);
      p/=ch[a[j]]+1;
      for(j=i+2;j<i+cnt-1;j++){
        sum+=s[j];
        up[a[i]]+=p*(sum+down[a[j]])*ch[a[j]]/(ch[a[j]]+1);
        p/=ch[a[j]]+1;
      }
      j=i+cnt-1;
      sum+=s[j];
      up[a[i]]+=p*(sum+down[a[j]]);
      //j<-i
      j=i+cnt-1;p=0.5;sum=s[j+1];
      up[a[i]]+=p*(sum+down[a[j]])*ch[a[j]]/(ch[a[j]]+1);
      p/=ch[a[j]]+1;
      for(j=i+cnt-2;j>i+1;j--){
        sum+=s[j+1];
        up[a[i]]+=p*(sum+down[a[j]])*ch[a[j]]/(ch[a[j]]+1);
        p/=ch[a[j]]+1;
      }
      j=i+1;
      sum+=s[j+1];
      up[a[i]]+=p*(sum+down[a[j]]);
    }
    for(i=1;i<=cnt;i++)dfs2(a[i],0);
  }
  for(i=1;i<=n;i++)ans+=(down[i]*ch[i]+up[i]*(d[i]-ch[i]))/d[i];
  printf("%.5f",ans/n);
  return 0;
}

  

 

posted @ 2014-05-13 11:01  Claris  阅读(670)  评论(0编辑  收藏  举报