BZOJ 1778: [Usaco2010 Hol]Dotp 驱逐猪猡 (高斯消元)

最新推荐文章于 2020-03-07 16:44:12 发布

原创最新推荐文章于 2020-03-07 16:44:12 发布 · 374 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#高斯消元 #概率DP

bzoj 同时被 3 个专栏收录

135 篇文章

订阅专栏

概率DP

9 篇文章

订阅专栏

高斯消元

2 篇文章

订阅专栏

本文介绍了一种使用高斯消元算法解决特定概率问题的方法，详细解析了如何建立转移方程并求解，同时讨论了精度问题及解决方案。

题面

题目传送门

分析

令爆炸概率为 $P$ 。设 $\large f(i)=\sum_{k=0}^{\infty}p_k(i)$ ， $p_k(i)$ 表示经过 $k$ 步走到 $i$ 的概率，那么在 $i$ 点结束的概率就为 $f (i) * P$ 。

看看 $f (i)$ 满足什么转移方程式。如下
$\large f(i)=\sum_{i-j}(f(j)*(1-P)/d_j)$
特别的，对于起点 $S$
$\large f(S)=\sum_{S-j}(f(j)*(1-P)/d_j)+1$

那么我们将左边移到右边，再把 $f (S)$ 的等式中 $+ 1$ 移到左边，就得到一个 $n$ 元方程组，高斯消元计算即可。

不知为什么原因WA？

这道题嘴上说着"误差不超过(1e-6)的答案会被接受"，但其实没有SPJ，必须输出九位小数，那么问题出现了，由于精度问题，高斯消元本该得到的答案为 $0$ ，但却得到了负零点几，那么直接输出就会输出"-0.000000000"，于是WA也。

所以我们要在输出时判断一下是不是小于(1e-9)就行了。

不过网上大多题解都是将等式右面往左边移，系数就全部取反了，这样也能过。不过为了避免输出-0，输出小数都还是特判一下吧。

CODE（左往右+特判）

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 305;
const double eps = 1e-15;
int n, m, A, B, d[MAXN];
double P, a[MAXN][MAXN];
inline void Guass(int N) {
	for(int j = 1; j <= N; ++j) {
		if(!a[j][j]) {
			for(int i = j+1; i <= N; ++i)
				if(a[i][j]) {
					for(int k = j; k <= N+1; ++k)
						swap(a[i][k], a[j][k]);
					break;
				}
		}
		for(int i = j+1; i <= N; ++i) {
			double v = a[i][j] / a[j][j];
			for(int k = j; k <= N+1; ++k)
				a[i][k] -= v*a[j][k];
		}
	}
	for(int i = N; i >= 1; --i) {
		for(int j = i+1; j <= N; ++j)
			a[i][N+1] -= a[j][N+1] * a[i][j];
		a[i][N+1] /= a[i][i];
	}
}

int main () {
	scanf("%d%d%d%d", &n, &m, &A, &B); P = (double)A/B;
	for(int i = 1, x, y; i <= m; ++i)
		scanf("%d%d", &x, &y), a[x][y] += 1, a[y][x] += 1, ++d[x], ++d[y];
	for(int i = 1; i <= n; ++i) {
		for(int j = 1; j <= n; ++j) {
			if(d[j]) a[i][j] /= d[j];
			a[i][j] *= (1-P);
		}
		a[i][i] -= 1;
	}
	a[1][n+1] = -1;
	Guass(n);
	for(int i = 1; i <= n; ++i)
		printf("%.9f\n", fabs(a[i][n+1]*P) < (1e-9) ? 0 : a[i][n+1]*P);
}

CODE2（右移左+无特判）

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 305;
const double eps = 1e-15;
int n, m, A, B, d[MAXN];
double P, a[MAXN][MAXN];
inline void Guass(int N) {
    for(int j = 1; j <= N; ++j) {
        if(!a[j][j]) {
            for(int i = j+1; i <= N; ++i)
                if(a[i][j]) {
                    for(int k = j; k <= N+1; ++k)
                        swap(a[i][k], a[j][k]);
                    break;
                }
        }
        for(int i = j+1; i <= N; ++i) {
            double v = a[i][j] / a[j][j];
            for(int k = j; k <= N+1; ++k)
                a[i][k] -= v*a[j][k];
        }
    }
    for(int i = N; i >= 1; --i) {
        for(int j = i+1; j <= N; ++j)
            a[i][N+1] -= a[j][N+1] * a[i][j];
        a[i][N+1] /= a[i][i];
    }
}
 
int main () {
    scanf("%d%d%d%d", &n, &m, &A, &B); P = (double)A/B;
    for(int i = 1, x, y; i <= m; ++i)
        scanf("%d%d", &x, &y), a[x][y] += 1, a[y][x] += 1, ++d[x], ++d[y];
    for(int i = 1; i <= n; ++i) {
        for(int j = 1; j <= n; ++j) {
            if(d[j]) a[i][j] /= d[j];
            a[i][j] *= (P-1);
        }
        a[i][i] += 1;
    }
    a[1][n+1] = 1;
    Guass(n);
    for(int i = 1; i <= n; ++i)
        printf("%.9f\n", a[i][n+1]*P);
}