数学知识——高斯消元

1. 算法用途
   求解多元线性方程组

2. 算法思路
   模拟线性代数中把系数矩阵从一般型化成最简型的过程

3. 具体步骤
   枚举每一列c
   1）找到绝对值最大的一行
   2）将该行换到最上面
   3）将该行的第一个数变成1
   4）把下面所有行的第c列消成0
   判断最终方程的个数和未知数的个数是否相等，如果不相等，且推出矛盾则无解，否则有无穷解。如果相等则有唯一解。
   有唯一解时从最后一行将值带到上面的行求出所有解

4. 例题：高斯消元解线性方程组
   输入一个包含n个方程n个未知数的线性方程组。
   方程组中的系数为实数。
   求解这个方程组。

下图为一个包含m个方程n个未知数的线性方程组示例：

输入格式
第一行包含整数n。
接下来n行，每行包含n+1个实数，表示一个方程的n个系数以及等号右侧的常数。

输出格式
如果给定线性方程组存在唯一解，则输出共n行，其中第i行输出第i个未知数的解，结果保留两位小数。
如果给定线性方程组存在无数解，则输出“Infinite group solutions”。
如果给定线性方程组无解，则输出“No solution”。

数据范围
1≤n≤100,
所有输入系数以及常数均保留两位小数，绝对值均不超过100。

输入样例：
3
1.00 2.00 -1.00 -6.00
2.00 1.00 -3.00 -9.00
-1.00 -1.00 2.00 7.00
输出样例：
1.00
-2.00
3.00

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
using namespace std;

const int N = 110;
const double eps = 1e-6;

int n;
double a[N][N];

int gauss()
{
    int c, r; // 列column. 行row
    for(c = 0, r = 0; c < n; c ++) //枚举每一列
    {
        int t = r;
        for(int i = r; i < n; i ++)
            if(fabs(a[i][c]) > fabs(a[t][c]))
                t = i; //找到第c列中绝对值最大的行
        
        if(fabs(a[t][c]) < eps)  continue;
        
        for(int i = c; i <= n; i ++)
            swap(a[t][i], a[r][i]); //将绝对值最大的一行换到第r行处
            
        for(int i = n; i >= c; i --) 
            a[r][i] /= a[r][c]; //将当前第r行的第一个数变成1
            
        for(int i = r + 1; i < n; i ++)
            if(fabs(a[i][c]) > eps)  //将第r行以下的所有行的第c列消成0
                for(int j = n; j >= c; j --)
                    a[i][j] -= a[i][c] * a[r][j];
        
        r ++;
    }
    
    if(r < n)
    {
        for(int i = r; i < n; i ++)
            if(fabs(a[i][n]) > eps)
                return 2; //无解(0 = 非零数)
        return 1; //无穷多组解
    }
    
    for(int i = n - 1; i >= 0; i --)
        for(int j = i + 1; j < n; j ++)
            a[i][n] -= a[i][j] * a[j][n];
            
    return 0;//有唯一解
}

int main()
{
    cin >> n;
    for(int i = 0; i < n; i ++)
        for(int j = 0; j <= n; j ++)
            cin >> a[i][j];
            
    int t = gauss();
    if(t == 0)
    {
        for(int i = 0; i < n; i ++)
            printf("%.2lf\n", a[i][n]);
    }
    else if(t == 1)  puts("Infinite group solutions");
    else puts("No solution");
    
    return 0;
}