高斯消元解线性方程组

#include <bits/stdc++.h>

using namespace std;

const int N = 110;
const double eps = 1e-7;
double a[N][N];
int n;

int gauss()
{
    int c, r;
    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;

          // 当前列全为零
        if (fabs(a[t][c]) < eps) continue;
        // 交换
        for (int i=c; i<=n; i++) swap(a[r][i], a[t][i]);
        // 当前行行首元素归一
        for (int i=n; i>=c; i--) a[r][i] /= a[r][c];

        // 用当前行消去下面所有行在当前列的数字
        for (int i = r+1; i < n; i++)
            if (fabs(a[i][c]) > eps) {
            // j 必须从n .. c，因为需要把 第c列放在最后减
                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; // 无解
        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()
{
	scanf("%d", &n);
	// n 个未知数
	// n 个方程
	// 输入形如 Ax + By + Cz = P, 输入的是 A B C P
	
	for(int i = 0; i < n; i++)
		for(int j = 0; j <= n; j ++)
			scanf("%lf", &a[i][j]);
	int res = gauss();
	if(res == 2) puts("No solution");
	else if(res == 1) puts("无数解");
	else
	{
		for(int i = 0; i < n; i ++) printf("%.2lf\n", a[i][n]);
	}
}