游乐园"/>
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->jj=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<-ij=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;
}
更多推荐
NOI2012 : 迷失游乐园
发布评论