luogu P4284 [SHOI2014]概率充电器

题面传送门
就这？就这？建议评蓝。
首先这个东西肯定是所有点的出现期望加起来。
考虑单个点怎么算，那么就是所有儿子的子树内不出现概率乘个边权乘起来最后用1减掉就好了。
那多个点不就换个根就好了。
时间复杂度\(O(n)\)
code:

#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define ll long long
#define db double
#define N 500000
#define mod 998244353
#define eps (1e-9)
#define U unsigned int
#define it iterator
#define Gc() getchar() 
#define Me(x,y) memset(x,y,sizeof(x))
#define d(x,y) (n*(x-1)+(y))
using namespace std;
int n,m,k,val[N+5],x,y,z;db F[N+5],dp[N+5],Ans;
struct yyy{int to,w,z;};
struct ljb{int head,h[N+5];yyy f[N+5<<1];I void add(int x,int y,int z){f[++head]=(yyy){y,z,h[x]};h[x]=head;}}s;
I void dfs1(int x,int last){
	dp[x]=1-val[x]/100.0;yyy tmp;for(int i=s.h[x];i;i=tmp.z)tmp=s.f[i],tmp.to^last&&(dfs1(tmp.to,x),dp[x]*=1-dp[tmp.to]*tmp.w/100.0);dp[x]=1-dp[x];
}
I void dfs2(int x,int last){
	Ans+=1-F[x];yyy tmp;for(int i=s.h[x];i;i=tmp.z) tmp=s.f[i],tmp.to^last&&(F[tmp.to]*=fabs(dp[tmp.to]*tmp.w/100.0-1)<eps?(0):(1-tmp.w/100.0*(1-F[x]/(1-dp[tmp.to]*tmp.w/100.0))),dfs2(tmp.to,x),0);
}
int main(){
	freopen("1.in","r",stdin);
	re int i;scanf("%d",&n);for(i=1;i<n;i++) scanf("%d%d%d",&x,&y,&z),s.add(x,y,z),s.add(y,x,z);for(i=1;i<=n;i++) scanf("%d",&val[i]);dfs1(1,0);for(i=1;i<=n;i++) F[i]=1-dp[i];dfs2(1,0); printf("%.6lf\n",Ans);
}

标签：struct,yyy,int,luogu,db,long,P4284,SHOI2014,define
来源： https://www.cnblogs.com/275307894a/p/15110266.html