高斯消元法

#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

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

int n;
double a[N][N]; // 系数矩阵

// 高斯消元法求解线性方程组

// 返回值：0 - 无解； 1 - 唯一解； 2 - 无穷解
int gauss()
{
	int row, col; // 当前操作行和列


	// 一列一列处理
	for (row = 0, col = 0; col < n; col++)
	{
		int max_row = row; // 找出当前处理列中绝对值最大的一行
		for (int i = row; i < n; i++)
		{
			if (fabs(a[i][col]) > fabs(a[max_row][col]))
			{
				max_row = i;
			}
		}

		// 如果找出的目标行的c列为0，无需处理了，直接进入下一轮循环
		if (fabs(a[max_row][col]) < eps)
		{
			continue;
		}

		// 将 max_row 行交换上来
		for (int i = col; i <= n; i++)
		{
			swap(a[max_row][i], a[row][i]);
		}

		// 将该行第一个数变为1，从后往前
		for (int i = n; i >= col; i--)
		{
			a[row][i] /= a[row][col];
		}

		// 将下面所有行的c列都消为0
		for (int i = row + 1; i < n; i++)
		{
			// 如果不为0，则消为零
			if (fabs(a[i][col]) > eps)
			{
				for (int j = n; j >= col; j--)
				{
					a[i][j] -= a[row][j] * a[i][col];
				}
			}
		}

		row++;
	}


	// 如果没有处理到所有行，说明不是唯一解
	if (row < n)
	{
		// 如果出现 0 = !0 ，出现矛盾，无解 
		for (int i = row; i < n; i++)
		{
			// 如果常系数不为0
			if (fabs(a[i][n]) > eps)
			{
				return 0; // 无解
			}
		}

		return 2; // 无穷多组解
	}


	// 已知有唯一解，求唯一解
	// 从后往前消
	for (int i = n - 1; i >= 0; i--)
	{
		for (int j = i + 1; j < n; j++)
		{
			a[i][n] -= a[j][n] * a[i][j];
			a[i][j] = 0;
		}
	}


	return 1; // 有唯一解
}




int main()
{
	cin >> n;

	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n + 1; j++)
		{
			cin >> a[i][j];
		}
	}

	int res = gauss();

	

	if (res == 0)
	{
		// 无解
		cout << "No solution" << endl;
	}
	else if (res == 1)
	{
		// 唯一解
		for (int i = 0; i < n; i++)
		{
			printf("%.2lf\n", a[i][n]);
		}
	}
	else if (res == 2)
	{
		// 无穷多个解
		cout << "Infinite group solutions" << endl;
	}

	return 0;
}