[学习笔记]省选算法·高斯消元Gauss

一、开头

神犇MX：大家好，我是神犇MX，每次都AK，喜欢d人。
xyz32768：大家好，我是蒟蒻xyz32768，每次都爆0，总是被d。
神犇MX：考下你，给定一个$n$个节点$m$条边的无向连通图，每到一个节点就会等概率随机地选择下一个与当前的点有边相连的节点走，求从$1$走到$n$的期望步数。
xyz32768：呃，$3$个节点，有边$(1,2)(2,3)$，有$\frac{1}{2}$的概率走$2$步，$\frac{1}{4}$的概率走$4$步，$\frac{1}{8}$的概率走$6$步，$\frac{1}{16}$的概率走$8$步，…，期望值为$\lim_{n\rightarrow\infty}\sum_{i=1}^n\frac{i}{2^{i-1}}=4$。
神犇MX：我说你xyz32768你真是菜啊，高斯消元一下不就行了！
xyz32768：什么？高斯消元？

二、概述

高斯消元是解$n$元一次方程组的算法。高斯消元的基本思想就是加减消元法。

三、预备

先将$n$个$n$元一次方程存进一个$n*(n+1)$的矩阵$a$，对于任意的$1\leq i,j \leq n,a[i][j]$表示第$i$个方程内，第$j$个未知数的系数，$a[i][n+1]$表示第$i$个方程等号右边的数值。举个例子：

$$\begin{cases} X_1-2X_2+3X_3=10\\ 2X_1+4X_2+7X_3=41\\ 13X_1-20X_2+15X_3=22 \end{cases}$$
其中$X_1,X_2,X_3$是未知数，那么储存的矩阵$a$就为：
$$\begin{bmatrix} 1 & -2 & 3 & 10\\ 2 & 4 & 7 & 41\\ 13 & -20 & 15 & 22 \end{bmatrix}$$

四、求解

分$n$步。每一步的操作都是把$a[i][i]$变为$1$，矩阵第$i$列的其他元素变为$0$（即将第$i$个方程的$X_i$系数变为$1$，其他方程的$X_i$系数变为$0$）。第$i$步的做法为：
1、将第$i$个方程的等式两边都除以$a[i][i]$，使第$i$个方程的$X_i$系数为$1$，也就是说，把矩阵第$i$行的所有元素除以$a[i][i]$。
2、消元，也就是把其他方程的$X_i$系数变为$0$。这时候就可以加减消元了，也就是说：
对于任何一个$1\leq j\leq n,j\neq i,i\leq k\leq n+1,$将$a[j][k]$减去$a[i][k]*a[j][i]$。
实现时，经常要在第$i$个到第$n$个方程中，使用$X_i$系数绝对值最大的方程作为第$i$个方程（将第$i$个元素绝对值最大的行与第$i$行交换），以避免浮点数计算的精度问题。
最后$n$步都结束后，所有的$a[i][i]$（对角线上的元素）都会变为$1$，除了对角线上的元素和第$n+1$列的元素之外，其他元素都为$0$，这时候方程的解就是第$n+1$列的元素。
特殊情况：如果矩阵的某一行出现$0$ $0$ $0$ $...$ $0$的情况，则方程有无穷多个解，如果出现$0$ $0$ $0$ $...$ $b$（$b\neq 0$），则方程无解。

五、实例

模拟下求解上面给出的方程：
初始状态：

$$\begin{bmatrix} 1 & -2 & 3 & 10\\ 2 & 4 & 7 & 41\\ 13 & -20 & 15 & 22 \end{bmatrix}$$
将 $X_1$的绝对值最大的方程置为第 $1$行：
$$\begin{bmatrix} 13 & -20 & 15 & 22\\ 2 & 4 & 7 & 41\\ 1 & -2 & 3 & 10 \end{bmatrix}$$
第 $1$个方程的$X_1$系数化为 $1$：
$$\begin{bmatrix} 1 & -\frac{20}{13} & \frac{15}{13} & \frac{22}{13}\\ 2 & 4 & 7 & 41\\ 1 & -2 & 3 & 10 \end{bmatrix}$$
将第 $2$行消元，即用$2X_1+4X_2+7X_3=41$减去 $2X_1-\frac{40}{13}X_2+\frac{30}{13}X_3=\frac{44}{13}$，矩阵变为：
$$\begin{bmatrix} 1 & -\frac{20}{13} & \frac{15}{13} & \frac{22}{13}\\ 0& \frac{92}{13} & \frac{61}{13} & \frac{489}{13}\\ 1 & -2 & 3 & 10 \end{bmatrix}$$
将第 $3$行消元，即
$$\begin{bmatrix} 1 & -\frac{20}{13} & \frac{15}{13} & \frac{22}{13}\\ 0& \frac{92}{13} & \frac{61}{13} & \frac{489}{13}\\ 0 & -\frac{6}{13} & \frac{24}{13} & \frac{108}{13} \end{bmatrix}$$
将 $X_2$的绝对值最大的方程置为第 $2$行并将系数化为$1$：
$$\begin{bmatrix} 1 & -\frac{20}{13} & \frac{15}{13} & \frac{22}{13}\\ 0& 1 & \frac{61}{92} & \frac{489}{92}\\ 0 & -\frac{6}{13} & \frac{24}{13} & \frac{108}{13} \end{bmatrix}$$
将第 $1$行和第$3$行消元：
$$\begin{bmatrix} 1 & 0 & \frac{50}{23} & \frac{227}{23}\\ 0& 1 & \frac{61}{92} & \frac{489}{92}\\ 0 & 0 & \frac{99}{46} & \frac{495}{46} \end{bmatrix}$$
第 $3$个方程系数化为$1$：
$$\begin{bmatrix} 1 & 0 & \frac{50}{23} & \frac{227}{23}\\ 0& 1 & \frac{61}{92} & \frac{489}{92}\\ 0 & 0 & 1 & 5 \end{bmatrix}$$
将 $1$、$2$行消元：
$$\begin{bmatrix} 1 & 0 & 0 & -1\\ 0& 1 & 0 & 2\\ 0 & 0 & 1 & 5 \end{bmatrix}$$
即 $X_1=-1,X_2=2,X_3=5$。

六、应用

高斯消元在竞赛中的部分用途如：

1、求解期望方程

求解期望方程等。举最简单的一个例子：给定一个无向连通图，每到一个节点就会等概率随机地选择下一个与当前节点有边相连的节点走，求从$1$走到$n$的期望步数。
这时，就可以设$X_i$为$i$到$n$的期望步数，其中$X_n=0$，那么容易得到，$X_i=\frac{1}{d_i}\sum(1+X_j)$（$d_i$为点$i$的度数，$j$为与$i$相邻的节点）
移项后就可以得到$n$个方程式。高斯消元后就能求出每个$X_i$了。

2、求解异或方程（补充中）

3、求矩阵的秩（补充中）

4、求解方阵的行列式

https://blog.csdn.net/xyz32768/article/details/81413569

5、求解可逆方阵的逆矩阵（补充中）

七、代码

高斯消元模板：

void Gauss() {
    int i, j, k;
    for (i = 1; i <= n; i++) {
        int u = i;
        for (j = i + 1; j <= n; j++)
            if (fabs(a[j][i]) > fabs(a[u][i])) u = j;
        if (u != i) for (j = i; j <= n + 1; j++)
            swap(a[i][j], a[u][j]);
        double tmp = a[i][i];
        for (j = i; j <= n + 1; j++) a[i][j] /= tmp;
        for (j = 1; j <= n; j++) if (j != i) {
            tmp = a[j][i];
            for (k = i; k <= n + 1; k++)
                a[j][k] -= a[i][k] * tmp;
        }
    }
}

八、题目

[BZOJ1013][JSOI2008]球形空间产生器sphere：
http://www.lydsy.com/JudgeOnline/problem.php?id=1013
[BZOJ2337][HNOI2011]XOR和路径：
http://www.lydsy.com/JudgeOnline/problem.php?id=2337
[BZOJ2707][SDOI2012]走迷宫：
http://www.lydsy.com/JudgeOnline/problem.php?id=2707
[BZOJ3143][HNOI2013]游走：
http://www.lydsy.com/JudgeOnline/problem.php?id=3143
[BZOJ4820][SDOI2017]硬币游戏：
http://www.lydsy.com/JudgeOnline/problem.php?id=4820