BZOJ3566 SHOI2014概率充電器(動態規劃+概率期望)
阿新 • • 發佈:2018-09-26
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; } } voidgetans(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概率充電器(動態規劃+概率期望)