[概率 树形DP] BZOJ 3566 [SHOI2014]概率充电器

题解：http://www.cnblogs.com/ctlchild/p/5105390.html

考虑补集转化。

先dfs一遍，令f[u]=(1-p[u])*∏(1-(1-f[v])*w) f[u]表示u这个点通过其子树并不能联通的概率。

然后考虑v从其父亲连过来的情况，设x=1-f[u]/(1-(1-f[v])*w)表示除掉v，u联通的概率，那么f[v]*=(1-x*w)

然后加起来就是答案了。

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define V G[p].v
#define eps 1e-8
using namespace std;

inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x)
{
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=500005;

struct edge{
	int u,v;
	double w;
	int next;
};

edge G[N*2];
int head[N],inum;

inline void add(int u,int v,double w,int p)
{
	G[p].u=u; G[p].v=v; G[p].w=w; G[p].next=head[u]; head[u]=p;
}

int n;
double q[N];
double ans,f[N];

inline void dfs(int u,int fa){
	f[u]=1-q[u];
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa)
		{
			dfs(V,u);
			f[u]*=1-((1-f[V])*G[p].w);
		}
}

inline void dp(int u,int fa){
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa)
		{
			double x=1-f[u]/(1-(1-f[V])*G[p].w);
			if (x>eps && f[V]>eps) f[V]*=(1-x*G[p].w);
			dp(V,u);
		}
}

int main()
{
	int iw,iu,iv;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	read(n); 
	for (int i=1;i<n;i++)
		read(iu),read(iv),read(iw),add(iu,iv,iw/100.0,++inum),add(iv,iu,iw/100.0,++inum);
	for (int i=1;i<=n;i++)
		read(iw),q[i]=iw/100.0;
	dfs(1,0);
	dp(1,0);
	for (int i=1;i<=n;i++)
		ans+=1-f[i];
	printf("%.6lf\n",ans);
	return 0;
}