牛客网NOIP赛前集训营-提高组（第三场）A-管道维修

A-管道维修

题目描述

在维修下水管道的过程中，发现一块n×m的易堵区域。为了方便表示，将易堵区域第$i$行第$j$列的格子命名为格子$(i,j)(1\le i\le n,1\le j\le m)$。每次维修下水道时，其中有些格子是堵塞状态，有些则是未堵塞状态。维修需要分步进行：所有与$n\times m$长方形四周边界相邻的格子或者与未被堵塞的格子相邻（四连通）的堵塞格子都可以在第1步内清理完毕。第k步过后，清理完毕的格子就变成未被堵塞的格子，这时与边界或者未堵塞的格子相邻的堵塞格子都会在第$k+1$步内被清理完毕。依此类推，直到所有格子都变成未堵塞的格子。现在下水道内的情况未知，格子$(i,j)$堵塞的概率为$p(i,j)$。不同格子之间堵塞的事件相互独立。现在求每个格子平均需要到第几步被清理完毕。一个格子如果一开始就是未堵塞的，算作在第$0$步被清理完毕。

输入描述:

第一行两个个正整数$n$和$m$表示区域的大小。$(1\le n,m\le 200)$ 接下来n行每行m个整数对$a,b$表示格子被堵塞的概率是$a/b$。$(0\le a\le b,1\le b<10^9+7)$

输出描述:

$n$行每行$m$个整数表示格子平均在第几步清理完毕。为了避免精度问题，如果平均在第$c/d$步内被清理完毕，则输出$c\cdot d-1(mod10^9+7)$，其中$d-1$表示$d$关于$10^9+7$的逆元，即$0$到$10^9+6$以内唯一一个和$d$相乘后模$10^9+7$等于$1$的整数。

示例1
输入

5 5
1 2 1 2 1 2 1 2 1 2
1 2 1 2 1 2 1 2 1 2
1 2 1 2 1 2 1 2 1 2
1 2 1 2 1 2 1 2 1 2
1 2 1 2 1 2 1 2 1 2

输出

500000004 500000004 500000004 500000004 500000004
500000004 781250006 781250006 781250006 500000004
500000004 781250006 665161138 781250006 500000004
500000004 781250006 781250006 781250006 500000004
500000004 500000004 500000004 500000004 500000004

说明

以左上的格子为例，如果这个格子是堵塞格子（发生概率为$1/2$），由于它和边界相邻，一定可以在第$1$轮就被清理完毕。如果它不是堵塞格子，按照定义在第$0$轮清理完毕。所以平均它在第$1/2$轮被清理完毕。$2$的逆元是$500000004$。其它与边界不相邻的格子有可能在第$2$轮才被清理完毕，取决于它们旁边的格子是否是堵塞的。

示例2
输入

3 3
1 2 1 3 1 2
1 3 2 3 1 3
1 2 1 3 1 2

输出

500000004 333333336 500000004
333333336 82304528 333333336
500000004 333333336 500000004

Solution

对于每一个$(i,j)$计算期望先考虑样例1
$(2,2)$的计算方法$\frac{1}{2}\times 0+\frac{1}{2}\times \frac{15}{16}\times 1+\frac{1}{2}\times \frac{1}{16}\times 2=\frac{17}{32}$
$17\times 32^{-1}\equiv 17\times 32^{1000000005}\equiv 781250006(\mathrm{m}\mathrm{o}\mathrm{d}1000000007)$
$(3,3)$的计算方法也类似$\frac{1}{2}\times 0+\frac{1}{2}\times \frac{15}{16}\times 1+\frac{1}{2}\times \frac{1}{16}\times \frac{255}{256}\times 2+\frac{1}{2}\times \frac{1}{16}\times \frac{1}{256}\times 3$（具体数值也没有算过）
可以发现一个规律:每层往外扩展都是分成两种情况
1.这层全是堵塞的(概率为$\prod \frac{a}{b}$)
2.这层不是全部堵塞(概率为$1-\prod \frac{a}{b}$)
其中第一种情况是可以继续扩展的
所以如果存在有一层$\exists a=0$,则$\prod \frac{a}{b}=0$可以直接跳出循环
后来维护每层的概率积有两种方法
1.暴力，时间复杂度$\theta (n^4)$
2.前缀积，时间复杂度$\theta (n^3)$,当然这个前缀积是要斜着维护的，而且要维护两个方向

这个前缀积还有坑！！！
如果有$0$的情况就会出现$\frac{0}{0}$的惨状，但是发现遇到$0$可以直接跳出，于是又用同样的方法维护了一遍$0$的个数，如果存在$0$就结束。
至于分数的运算很麻烦的问题…dalao告诉我们可以直接使用逆元进行运算，不会有任何问题

Code(然而不知道哪出bug了，94分…)

#include <cstdio>
#include <algorithm>
#define N 205
#define MOD 1000000007

typedef long long LL;
using namespace std;

int ll[N][N], rr[N][N];//ll,rr是0的个数前缀和
LL lo[N][N], ro[N][N], val[N][N];//lo,ro是前缀积,val是一个点的概率的逆元

LL qui_pow(LL x, int y) {
	if (y == 1) return x;
	LL t = qui_pow(x, y / 2);
	if (y % 2 == 0) return t * t % MOD;
	else return t * t % MOD * x % MOD;
}

int main() {
	int n, m;
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; ++i) {
		for (int j = 1; j <= m; ++j) {
			LL a, b;
			scanf("%lld%lld", &a, &b);
			val[i][j] = a * qui_pow(b, MOD - 2) % MOD;
		}
	}
	for (int i = 0; i <= n + 1; ++i) {
		for (int j = 0; j <= n + 1; ++j) {
			if (i == 0 || j == 0) {
				lo[i][j] = ro[i][j] = 1;//不能出现lo,ro为零的情况
			}
		}
	}
	for (int i = 1; i <= n; ++i) {//前缀预处理
		for (int j = 1; j <= m; ++j) {
			if (val[i][j] == 0) {
				ll[i][j] = ll[i - 1][j - 1] + 1;
				rr[i][j] = rr[i - 1][j + 1] + 1;
			}
			else {
				ll[i][j] = ll[i - 1][j - 1];
				rr[i][j] = rr[i - 1][j + 1];
			}
			lo[i][j] = lo[i - 1][j - 1] * val[i][j] % MOD;
			ro[i][j] = ro[i - 1][j + 1] * val[i][j] % MOD;
			if (lo[i][j] == 0) lo[i][j] = 1;
			if (ro[i][j] == 0) ro[i][j] = 1;//不能出现lo,ro为零的情况
		}
	}
	for (int i = 1; i <= n; ++i) {
		for (int j = 1; j <= m; ++j) {
			int p = min(min(i, j), min(n - i + 1, m - j + 1));//最大层数
			LL ans = 0, now = val[i][j];
			for (int k = 1; k <= p; ++k) {
				int cnt = ll[i + k - 1][j] - ll[i - 1][j - k]
						+ ll[i][j + k - 1] - ll[i - k][j - 1]
						+ rr[i - 1][j - k + 2] - rr[i - k + 1][j] 
						+ rr[i + k - 2][j + 1] - rr[i][j + k - 1];//0个数
				LL  pob = lo[i + k][j] * qui_pow(lo[i - 1][j - k - 1], MOD - 2) % MOD
						* lo[i][j + k] % MOD * qui_pow(lo[i - k - 1][j - 1], MOD - 2) % MOD
						* ro[i - 1][j - k + 1] % MOD * qui_pow(ro[i - k][j], MOD - 2) % MOD
						* ro[i + k - 1][j + 1] % MOD * qui_pow(ro[i][j + k], MOD - 2) % MOD;//概率的逆元
				if (cnt > 0) {//处理边界小问题，break
					ans = (ans + now * (k - 1)) % MOD;
					break;
				}
				if (k < p) ans = (ans + (MOD + 1 - pob) * now % MOD * k) % MOD;
				else ans = (ans + now * k) % MOD;//处理边界小问题同上
				now = pob * now % MOD;
			}
			printf("%lld", ans);
			if (j < m) putchar(' ');
			else putchar('\n');
		}
	}
	return 0;
}

蒟蒻只能码提高t1了qnq