• BZOJ 3566: [SHOI2014]概率充电器


    简单的概率DP,思路极其顺畅(然后刚开始还是手残WA了两发)

    首先由于这里的每个点贡献都是(1),因此期望和就是概率和,换句话说我们要求(sum_{i=1}^n P(i))(P(i))表示(i)被点亮的概率

    考虑一个点被点亮的情况,要么是自己亮要么是别人送电给它亮

    (T(P(X),P(Y)))表示事件(X,Y)中至少有一件发生的概率,讨论一下四种情况后得到(T(P(X),P(Y))=P(X)+P(Y)-P(X) imes P(Y)),则我们有:

    [P(x)=T(q_x,P(y) imes E_{x,y}) (x,y)in E ]

    woc这式子一看,妈妈我会高斯消元!然后看一眼数据范围,可以得到0分的好成绩……

    但是我们仔细一想,这个式子是针对所有图的,然而题目中的是棵树,而树的特点就是无环,转移单一

    容易发现在树上一个点被送电的情况只有两种:从子树内来和子树外来(如果我们把它定义成有根树的话)

    然后是不是豁然开朗,这就是一个裸的换根DP的套路,我们先一遍求出只从自己子树内转移来的概率

    然后再一遍DFS求出从子树外来的概率,这部分可以用父亲的概率消除掉自己转移到父亲的那一部分来计算,直接把上面的式子移个项就能解出来了,注意分母为(0)的情况特别考虑

    #include<cstdio>
    #define RI register int
    #define CI const int&
    using namespace std;
    const int N=500005;
    struct edge
    {
    	int to,nxt; double p;
    }e[N<<1]; int head[N],n,cnt,x,y,z; double p[N],up[N],dn[N],ans;
    inline void addedge(CI x,CI y,const double& z)
    {
    	e[++cnt]=(edge){y,head[x],z}; head[x]=cnt;
    	e[++cnt]=(edge){x,head[y],z}; head[y]=cnt;
    }
    inline double T(const double& x,const double& y)
    {
    	return x+y-x*y;
    }
    #define to e[i].to
    inline void DFS1(CI now=1,CI fa=0)
    {
    	for (RI i=head[now];i;i=e[i].nxt)
    	if (to!=fa) DFS1(to,now),dn[now]=T(dn[now],T(p[to],dn[to])*e[i].p);
    }
    inline void DFS2(CI now=1,CI fa=0,const double& ep=0)
    {
    	if (fa)
    	{
    		up[now]=T(p[fa],up[fa]); double pn=ep*T(p[now],dn[now]);
    		if (pn!=1.0) up[now]=T(up[now],(dn[fa]-pn)/(1.0-pn)); up[now]*=ep;
    	}
    	for (RI i=head[now];i;i=e[i].nxt) if (to!=fa) DFS2(to,now,e[i].p);
    }
    #undef to
    int main()
    {
    	RI i; for (scanf("%d",&n),i=1;i<n;++i)
    	scanf("%d%d%d",&x,&y,&z),addedge(x,y,0.01*z);
    	for (i=1;i<=n;++i) scanf("%d",&x),p[i]=0.01*x;
    	for (DFS1(),DFS2(),i=1;i<=n;++i) ans+=T(p[i],T(up[i],dn[i]));
    	return printf("%.6lf",ans),0;
    }
    
  • 相关阅读:
    [原]Android 开发第一步
    [转]使用Android-Studio 开发Android 程序
    [转]VS2010 常用插件
    [转]FluentData
    BUUCTF-[HCTF 2018]WarmUp
    2019.11.11读书笔记
    2019.11.9读书笔记
    记录一道神仙CTF-wtf.sh-150
    SDOI2018 一轮培训划水祭
    [SHOI2009]会场预约
  • 原文地址:https://www.cnblogs.com/cjjsb/p/12261494.html
Copyright © 2020-2023  润新知