首页 > 代码库 > 【bzoj1778】[Usaco2010 Hol]Dotp 驱逐猪猡 矩阵乘法+概率dp+高斯消元

【bzoj1778】[Usaco2010 Hol]Dotp 驱逐猪猡 矩阵乘法+概率dp+高斯消元

题目描述

奶牛们建立了一个随机化的臭气炸弹来驱逐猪猡。猪猡的文明包含1到N (2 <= N <= 300)一共N个猪城。这些城市由M (1 <= M <= 44,850)条由两个不同端点A_j和B_j (1 <= A_j<= N; 1 <= B_j <= N)表示的双向道路连接。保证城市1至少连接一个其它的城市。一开始臭气弹会被放在城市1。每个小时(包括第一个小时),它有P/Q (1 <= P <=1,000,000; 1 <= Q <= 1,000,000)的概率污染它所在的城市。如果这个小时内它没有污染它所在的城市,那麽它随机地选择一条道路,在这个小时内沿着这条道路走到一个新的城市。可以离开这个城市的所有道路被选择的概率均等。因为这个臭气弹的随机的性质,奶牛们很困惑哪个城市最有可能被污染。给定一个猪猡文明的地图和臭气弹在每个小时内爆炸的概率。计算每个城市最终被污染的概率。如下例,假设这个猪猡文明有两个连接在一起的城市。臭气炸弹从城市1出发,每到一个城市,它都有1/2的概率爆炸。 1--2 可知下面这些路径是炸弹可能经过的路径(最后一个城市是臭气弹爆炸的城市): 1: 1 2: 1-2 3: 1-2-1 4: 1-2-1-2 5: 1-2-1-2-1 ... 要得到炸弹在城市1终止的概率,我们可以把上面的第1,第3,第5……条路径的概率加起来,(也就是上表奇数编号的路径)。上表中第k条路径的概率正好是(1/2)^k,也就是必须在前k-1个回合离开所在城市(每次的概率为1 - 1/2 = 1/2)并且留在最后一个城市(概率为1/2)。所以在城市1结束的概率可以表示为1/2 + (1/2)^3 + (1/2)^5 + ...。当我们无限地计算把这些项一个个加起来,我们最后会恰好得到2/3,也就是我们要求的概率,大约是0.666666667。这意味着最终停留在城市2的概率为1/3,大约为0.333333333。

输入

* 第1行: 四个由空格隔开的整数: N, M, P, 和 Q * 第2到第M+1行: 第i+1行用两个由空格隔开的整数A_j和B_j表示一条道路。

输出

* 第1到第N行: 在第i行,用一个浮点数输出城市i被摧毁的概率。误差不超过10^-6的答桉会 被接受(注意这就是说你需要至少输出6位有效数字使得答桉有效)。

样例输入

2 1 1 2
1 2

样例输出

0.666666667
0.333333333


题解

矩阵乘法+概率dp+高斯消元

%PoPoQQQ,大爷就是强啊

设S为初始炸弹出现在每个位置的概率,那么显然S=[1,0,0,...,0]

设T为炸弹移动的矩阵,T[i][j]表示从i不爆炸且移动到j的概率,那么如果i到j有边,则T[i][j]=(1-p/q)/d[i],否则T[i][j]=0,易知T[i][j]<1-p/q。

根据定义,很容易得知(T*T)[i][j]<(1-p/q)^2,(T*T*T)[i][j]<(1-p/q)^3,...,$T^\infty$[i][j]<$(1-p/q)^\infty$=0。

而炸弹在第1s内爆炸的概率为$\frac pqS$,第2s内爆炸的概率为$\frac pqST$,第3s内爆炸的概率为$\frac pqST^2$,...

故答案ANS(炸弹在每个位置爆炸的概率)有如下关系:

技术分享

其中,I表示单位矩阵,第一个式子使用了等比数列求和公式,而第二个式子将已知矩阵I-T乘到左端。

然后我们可以发现I-T、p/q*S都是已知矩阵,只有ANS未知,并且ANS是一个1*n的矩阵,只有n个未知数。

所以我们可以设这些未知数分别为x1、x2、...、xn,然后根据已知条件列出方程,使用高斯消元求解。

#include <cstdio>#include <cmath>#include <algorithm>#define N 310#define M 90010using namespace std;int x[M] , y[M] , d[N] , n;double a[N][N];void gauss(){	int i , j , k;	double t;	for(i = 1 ; i <= n ; i ++ )	{		for(k = 0 , j = i ; j <= n ; j ++ )			if(fabs(a[j][i]) > fabs(a[k][i]))				k = j;		for(j = i ; j <= n + 1 ; j ++ ) swap(a[i][j] , a[k][j]);		for(j = i + 1 ; j <= n ; j ++ )			for(t = a[j][i] / a[i][i] , k = i ; k <= n + 1 ; k ++ )				a[j][k] -= t * a[i][k];	}	for(i = n ; i >= 1 ; i -- )	{		for(j = i + 1 ; j <= n ; j ++ ) a[i][n + 1] -= a[i][j] * a[j][n + 1];		a[i][n + 1] /= a[i][i];	}}int main(){	int m , p , q , i;	scanf("%d%d%d%d" , &n , &m , &p , &q);	for(i = 1 ; i <= m ; i ++ ) scanf("%d%d" , &x[i] , &y[i]) , d[x[i]] ++ , d[y[i]] ++ ;	for(i = 1 ; i <= m ; i ++ ) a[x[i]][y[i]] = ((double)p / q - 1) / d[y[i]] , a[y[i]][x[i]] = ((double)p / q - 1) / d[x[i]];	for(i = 1 ; i <= n ; i ++ ) a[i][i] = 1;	a[1][n + 1] = (double)p / q;	gauss();	for(i = 1 ; i <= n ; i ++ ) printf("%.9lf\n" , a[i][n + 1]);	return 0;}

 

【bzoj1778】[Usaco2010 Hol]Dotp 驱逐猪猡 矩阵乘法+概率dp+高斯消元