[TopCoder 12984] TorusSailing（高斯消元主元法優化）

題面

Vjudge TorusSailing

分析

首先得到一個DP方程$dp[i][j]=\dfrac{1}{2}\left(dp[(i + 1)\ \text{mod}\ N][j]+dp[i][(j + 1)\ \text{mod}\ M]\right)+1\quad\quad①$高消$\sout{O(N^6)}$走人。

不能直接高消，我就想按照CF963E Circles of Waiting的做法，由於有很多係數爲$0$，那就標記一下不爲零的地方，亂搞一下把高消變成$O(k\cdot n)$的。經過超過2小時的嘗試，我證明了自己還是太菜太naive。

由於這個環面上可以隨意“平移”，不妨把$(goalX,goalY)$放到$(N-1,M-1)$的地方，那麼我們的問題就變成了求$(N-1-goalX, M-1-goalY)$到$(N-1,M-1)$的步數期望。

我們把第$N-1$行和第$M-1$列的$dp$值作爲主元，根據$①$式，我們可以用這些主元由下至上，由右至左表示出其他的$dp$值，然後再由$①$式將第$0$行、第$0$列、第$N - 1$行、第$M - 1$列之間的關係表示成一個$N+M-1$元$1$次方程組，用高斯消元解這個方程組，時間複雜度$O(N^3)$，進而可以知道所有$dp$值。

代碼

點$(N - 1, M - 1)$不要處理，否則會算重。如果訪問到$(N - 1, M - 1)$說明起點終點重合，答案是$0$。

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <vector>

const int MAXN = 100;

struct Node {
	double A[MAXN * 2 + 5], B;
}P[MAXN + 5][MAXN + 5];

double Abs(double x) {
	return x > 0 ? x : -x;
}

double A[MAXN * 2 + 5][MAXN * 2 + 5], Res[MAXN * 2 + 5];
	
void Gauss(int N) {
	for (int i = 0; i < N; i++) {
		double tmp = A[i][i];
		for (int j = i; j <= N; j++)
			A[i][j] /= tmp;
		for (int j = i + 1; j < N; j++) {
			tmp = A[j][i];
			for (int k = i; k <= N; k++)
				A[j][k] -= tmp * A[i][k];
		}
	}
	for (int i = N - 1; i >= 0; i--) {
		Res[i] = A[i][N];
		for (int j = i + 1; j < N; j++)
			Res[i] -= A[i][j] * Res[j];
	}
}

class TorusSailing {
	public:
	double expectedTime(int N, int M, int X, int Y) {
		for (int i = 0; i < N - 1; i++)
			P[i][M - 1].A[i] = 1;
		for (int i = 0; i < M - 1; i++)
			P[N - 1][i].A[i + N - 1] = 1;
		for (int i = N - 2; i >= 0; i--)
			for (int j = M - 2; j >= 0; j--) {
				P[i][j].B = 0.5 * (P[i][j + 1].B + P[i + 1][j].B) + 1;
				for (int k = 0; k < N + M - 2; k++)
					P[i][j].A[k] = 0.5 * (P[i][j + 1].A[k] + P[i + 1][j].A[k]);
			}
		for (int i = 0; i < N - 1; i++) {
			A[i][i] = 1, A[i][N + M - 2] = 0.5 * P[i][0].B + 1;
			for (int k = 0; k < N + M - 2; k++)
				A[i][k] -= 0.5 * (P[i][0].A[k] + P[i + 1][M - 1].A[k]);
		}
		for (int i = 0; i < M - 1; i++) {
			int j = i + N - 1;
			A[j][j] = 1, A[j][N + M - 2] = 0.5 * P[0][i].B + 1;
			for (int k = 0; k < N + M - 2; k++)
				A[j][k] -= 0.5 * (P[0][i].A[k] + P[N - 1][i + 1].A[k]);
		}
		Gauss(N + M - 2);
		X = N - 1 - X, Y = M - 1 - Y;
		double Ans = P[X][Y].B;
		for (int i = 0; i < N + M - 2; i++)
			Ans += Res[i] * P[X][Y].A[i];
		return Ans;
	}
};