【BZOJ】3640: JC的小苹果

题解

我们考虑列出期望方程组，(dp[i][j])表示在第(i)个点血量为(j)的时候到达(N)点的概率，所有的(dp[N][j])都是1，所有(j < 0)都是0
答案是(dp[1][hp])

(dp[u][j] = sum_{v} frac{1}{deg[u]}dp[v][j - a[v]])
我们发现这个方程在j不同的时候，只有常数项发生改变，剩下的系数不变

于是我们把常数项变成一个列向量，把需要消元的系数矩阵求一个逆，每次计算常数项，用常数项乘逆矩阵得到每一个血量的(dp)值即可

注意判自环= =

代码

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define pdi pair<db,int>
#define mp make_pair
#define pb push_back
#define enter putchar('
')
#define space putchar(' ')
#define eps 1e-8
#define mo 974711
#define MAXN 100005
//#define ivorysi
using namespace std;
typedef long long int64;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;char c = getchar();T f = 1;
    while(c < '0' || c > '9') {
	if(c == '-') f = -1;
	c = getchar();
    }
    while(c >= '0' && c <= '9') {
	res = res * 10 + c - '0';
	c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) {
	out(x / 10);
    }
    putchar('0' + x % 10);
}
int N,M,hp,deg[155],blood[155];
struct node {
    int to,next;
}E[100005];
int sumE,head[155];
db b[155],dp[155][10005],c[155];
void add(int u,int v) {
    E[++sumE].next = head[u];
    E[sumE].to = v;
    head[u] = sumE;
}
struct Matrix {
    db f[155][155];
    Matrix() {memset(f,0,sizeof(f));}
    void unit() {
	for(int i = 1 ; i <= N ; ++i) {
	    f[i][i] = 1.0;
	}
    }
    friend Matrix operator * (const Matrix &a,const Matrix &b) {
	Matrix c;
	for(int i = 1 ; i <= N ; ++i) {
	    for(int j = 1 ; j <= N ; ++j) {
		for(int k = 1 ; k <= N ; ++k) {
		    c.f[i][j] += a.f[i][k] * b.f[k][j];
		}
	    }
	}
	return c;
    }
    friend Matrix operator ~(Matrix a) {
	Matrix b;
	b.unit();
	for(int i = 1 ; i <= N ; ++i) {
	    int l = i;
	    for(int j = i + 1; j <= N ; ++j) {
		if(fabs(a.f[j][i]) > fabs(a.f[l][i])) l = j;
	    }
	    if(i != l) {
		for(int j = 1 ; j <= N ; ++j) {
		    swap(a.f[i][j],a.f[l][j]);
		    swap(b.f[i][j],b.f[l][j]);
		}
	    }
	    db t = 1.0 / a.f[i][i];
	    for(int j = 1 ; j <= N ; ++j) {
		a.f[i][j] *= t;
		b.f[i][j] *= t;
	    }
	    
	    for(int j = 1 ; j <= N ; ++j) {
		if(i == j) continue;
		db t = a.f[j][i];
		for(int k = 1 ; k <= N ; ++k) {
		    a.f[j][k] -= t * a.f[i][k];
		    b.f[j][k] -= t * b.f[i][k];
		}
	    }
	}
	return b;
    }
}A;
void Solve() {
    read(N);read(M);read(hp);
    int u,v;
    for(int i = 1 ; i <= N ; ++i) read(blood[i]);
    for(int i = 1 ; i <= M ; ++i) {
	read(u);read(v);
	if(u == v) {add(u,v);++deg[u];}
	else {add(u,v);add(v,u);++deg[u];++deg[v];}
    }
    for(int u = 1 ; u <= N ; ++u) {
	if(u == N) {
	    b[u] = 1;A.f[u][u] = 1;continue;
	}
	A.f[u][u] = 1;
	for(int i = head[u] ; i ; i = E[i].next) {
	    int v = E[i].to;
	    if(v == N) {b[u] += 1.0 / deg[u];}
	    else if(blood[v] == 0){
		A.f[v][u] -= 1.0 / deg[u];
	    }
	}
    }
    A = ~A;
    for(int i = 1 ; i <= hp ; ++i) {
	for(int u = 1 ; u <= N ; ++u) {
	    c[u] = b[u];
	    if(u == N) continue;
	    for(int h = head[u] ; h ; h = E[h].next) {
		int v = E[h].to;
		if(blood[v] && i > blood[v]) {
		    c[u] += (1.0 / deg[u]) * dp[v][i - blood[v]];
		}
	    }
	}
	for(int u = 1 ; u <= N ; ++u) {
	    for(int k = 1 ; k <= N ; ++k) {
		dp[u][i] += c[k] * A.f[k][u];
	    }
	}
    }
    printf("%.8lf
",dp[1][hp]);
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Solve();
    return 0;
}