http://acm.hdu.edu.cn/showproblem.php?pid=4035
题意:给一棵n个节点的树,每个节点有值k[i]和e[i],分别表示k[i]概率走向1号节点,e[i]概率获得胜利(即停止),如果没有进行上边任意操作,则等概率的走向与这个节点连边的点。问走过的边的期望。(n<=10000)
#include <cstdio>
#include <cstring>
using namespace std;
const int N=10005;
const double eps=1e-10;
struct dat { double a, b, c, K, E; }D[N];
struct E { int next, to; }e[N<<1];
int n, cnt, ihead[N];
void clr() {
memset(ihead, 0, sizeof(int)*(n+1));
cnt=0;
}
void add(int u, int v) {
e[++cnt].next=ihead[u]; ihead[u]=cnt; e[cnt].to=v;
e[++cnt].next=ihead[v]; ihead[v]=cnt; e[cnt].to=u;
}
double abs(double x) { return x<0?-x:x; }
void dfs(int x, int fa) {
bool flag=1;
double temp=1-D[x].K-D[x].E, suma=0, sumb=0, sumc=0;
int m=0;
for(int i=ihead[x]; i; ++m, i=e[i].next) if(e[i].to!=fa) {
flag=0;
dfs(e[i].to, x);
suma+=D[e[i].to].a;
sumb+=D[e[i].to].b;
sumc+=D[e[i].to].c;
}
if(flag) { D[x].a=D[x].K; D[x].b=D[x].c=temp; return; }
double di=m-temp*sumb;
D[x].a=(D[x].K*m+temp*suma)/di;
D[x].b=temp/di;
D[x].c=temp*(sumc+m)/di;
}
void work() {
dfs(1, -1);
if(abs(1-D[1].a)<=eps) puts("impossible");
else printf("%.6f\n", D[1].c/(1-D[1].a));
}
int main() {
int T; scanf("%d", &T);
for(int TT=1; TT<=T; ++TT) {
scanf("%d", &n);
for(int i=0; i<n-1; ++i) { int x, y; scanf("%d%d", &x, &y); add(x, y); }
for(int i=1; i<=n; ++i) scanf("%lf%lf", &D[i].K, &D[i].E), D[i].K/=100, D[i].E/=100;
printf("Case %d: ", TT);
work();
clr();
}
return 0;
}
好神的题...做出这题感觉对期望的认识大大加深了呢...
设状态$E[i]$表示当前在$i$节点还需要走$E[i]$的期望边即可结束,答案就是$E[1]$,容易得到:
对于叶子节点:
$$
\begin{align}
E[i] & = k[i]E[1]+e[i]*0+(1-k[i]-e[i])*(E[父亲]+1) \\
& = k[i]E[1]+(1-k[i]-e[i])E[父亲]+(1-k[i]-e[i])
\end{align}
$$
对于非叶子节点:
$$
\begin{align}
E[i] & = k[i]E[1]+e[i]*0+\frac{(1-k[i]-e[i])(E[父亲]+1+\sum_{j是i孩子} (E[j]+1)}{m} \\
& = k[i]E[1]+\frac{1-k[i]-e[i]}{m}E[父亲] + \frac{1-k[i]-e[i]}{m} \sum_{j是i孩子} E[j] + (1-k[i]-e[i])
\end{align}
$$
由于是树,我们要深度挖掘他们的性质:
1、我们发现非叶节点里只是多了$\sum_{j是i孩子} E[j]$
2、由于期望本来就是求极限意义,如果裸的这样求显然是递归无限的,我们需要通过数学技巧约掉式子
比如我们可以为了方便,设$E[i]$的一般式子为$E[i]=A[i]E[1]+B[i]E[父亲]+C[i]$(为什么带个父亲未知数呢?那是因为$i$的儿子需要知道自己父亲(即$i$)的期望,那么我们可以通过带入到原式中且移项即可解出$E[i]$,是不是很神奇呢0.0)
那么$\sum_{j是i孩子} E[j] = \sum_{j是i孩子} \left( A[j]E[1] + B[j]E[i] + C[j] \right)$
最终化简超强式子,变为求出这个一般方程的系数!:
对于叶子节点:
$$A[i]=k[i], B[i]=C[i]=1-k[i]-e[i]$$
对于非叶子节点:
设$di=m-(1-k[i]-e[i])\sum_{j} B[j], m=所有与i连边的节点数$
$$
\begin{align}
A[i] & = (k[i]m+(1-k[i]-e[i])\sum_{j} A[j]) / di \\
B[i] & = (1-k[i]-e[i]) / di \\
C[i] & = (1-k[i]-e[i])(m+\sum_{j} C[j]) / di
\end{align}
$$
最后答案:
$E[1]=C[i]/(1-A[1])$,判一下除数是否为0即可判断有无解。