1. 程式人生 > >BZOJ3566 SHOI2014概率充電器(動態規劃+概率期望)

BZOJ3566 SHOI2014概率充電器(動態規劃+概率期望)

pri efi struct lin clas bzoj3566 mat == spa

  設f[i]為i在子樹內不與充電點連通的概率。則f[i]=(1-pi)·∏(1-qk+qk·f[k])。

  然後從父親更新答案。則f[i]=f[i]·(1-qfa+qfa*f[fa]/(1-qfa+qfa*f[i]))。

  比較好想的dp。註意第二個式子可能會除0。

#include<iostream> 
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read() { int x=0,f=1;char c=getchar(); while (c<0||c>9) {if (c==-) f=-1;c=getchar();} while (c>=0&&c<=9) x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } #define N 500010 int n,P[N],p[N],t=0; double f[N],ans=0; struct data{int to,nxt,len; }edge[N
<<1]; void addedge(int x,int y,int z){t++;edge[t].to=y,edge[t].nxt=p[x],edge[t].len=z,p[x]=t;} void dfs(int k,int from) { f[k]=1-P[k]/100.0; for (int i=p[k];i;i=edge[i].nxt) if (edge[i].to!=from) { dfs(edge[i].to,k); f[k]*=(f[edge[i].to]-1)*edge[i].len/100+1; } } void
getans(int k,int from) { for (int i=p[k];i;i=edge[i].nxt) if (edge[i].to!=from) { double q=edge[i].len/100.0; if (f[edge[i].to]>0) f[edge[i].to]*=1-q+q*f[k]/(1-q+q*f[edge[i].to]); getans(edge[i].to,k); } } int main() { #ifndef ONLINE_JUDGE freopen("bzoj3566.in","r",stdin); freopen("bzoj3566.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(); for (int i=1;i<n;i++) { int x=read(),y=read(),z=read(); addedge(x,y,z),addedge(y,x,z); } for (int i=1;i<=n;i++) P[i]=read(); dfs(1,1); getans(1,1); for (int i=1;i<=n;i++) ans+=1-f[i]; printf("%.6lf",ans); return 0; }

BZOJ3566 SHOI2014概率充電器(動態規劃+概率期望)