题目大意:一颗树,给定每个点消失的概率,求出连通块的期望值。要求支持修改消失概率的操作并且给出每次修改过后的期望值。注意被破坏的点不能算入连通块中。
数据范围,时限1S。
传送门 D. Interstellar battle
我们考虑做有根树的DP。设1为根。
我们设为v节点消失的概率,设分别表示v节点被破坏/没被破坏时的连通块期望值。
解释一下f[v][1]的转移方程:因为如果v节点没有被破坏,并且儿子节点sn也没有被破坏,那么连通块的个数会减少,减少的数量就是sn所在的连通块的期望,也就是。
当然我们不可能每次询问了就DFS一遍计算,所以我们需要再研究一下递推式。
我们先只考虑4号点对答案的贡献。我么按照递推式模拟一遍。
最后答案就是,也就是。推广到一般情况:v对答案的贡献就是,特别地,设(0是1的父亲)。
知道这个结论过后维护起来就特别方便了。我们记。修改一个点的概率时就相应地修改值就行了(具体见代码)。
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<ctime>
#define ll long long
#define N 100005using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}int n,m;
double p[N];
struct load {int to,next;}s[N<<1];
int h[N],cnt;
void add(int i,int j) {s[++cnt]=(load) {j,h[i]};h[i]=cnt;}
int fa[N];
double sum[N];
void dfs(int v,int fr) {for(int i=h[v];i;i=s[i].next) {int to=s[i].to;if(to==fr) continue ;fa[to]=v;dfs(to,v);sum[v]+=(1-p[to]);}
}
double ans,c;
int main() {n=Get();for(int i=1;i<=n;i++) scanf("%lf",&p[i]);p[0]=1;int a,b;for(int i=1;i<n;i++) {a=Get()+1,b=Get()+1;add(a,b),add(b,a);}dfs(1,0);for(int i=1;i<=n;i++) {ans+=p[fa[i]]*(1-p[i]);}m=Get();while(m--) {a=Get()+1;scanf("%lf",&c);ans-=p[fa[a]]*(1-p[a]);ans-=sum[a]*p[a];sum[fa[a]]-=1-p[a];p[a]=c;ans+=p[fa[a]]*(1-p[a]);ans+=sum[a]*p[a];sum[fa[a]]+=1-p[a];cout<<ans<<"\n";}return 0;
}
然而,总觉得我的做法太不清真了。网上一搜题解,才发现了自己的naive。原来期望是可以分开计算的,也就是说可以分别算出每一对节点对答案的贡献在加起来。然后公式基本相同的。