高斯消元解线性方程组

题目

输入一个包含 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<cmath>
#include<cstdio>
 
using namespace std;
 
const int N = 110;
double s[N][N];//矩阵 
 
const double eps = 1e-6;
int n;
 
int gauss()
{
	int r, c;
	for(r = 0, c = 0; c < n; c ++)
	{
		int t = r;//找到绝对值最大的一行
		for(int i = r + 1; i < n; i ++)
			if(fabs(s[t][c]) < fabs(s[i][c]))
				t = i; 
		
		if(fabs(s[t][c]) < eps) continue;//全为0 
		
		for(int i = c; i <= n; i ++) swap(s[t][i], s[r][i]);//换到第一行
		
		for(int i = n; i >= c; i --) s[r][i] /= s[r][c];//把第一个数消成1
		 
		for(int i = r + 1; i < n; i ++)//将下面所有行的第c列消成0
			if(fabs(s[i][c]) > eps)
				for(int j = n; j >= c; j --)
					s[i][j] -= s[r][j] * s[i][c];
					
		r ++;
	}
	
	if(r < n)
	{
		for(int i = r; i < n; i ++)
			if(fabs(s[i][n]) > eps)
				return 2;//无解
		return 1; 
	}
	
	for(int i = n - 1; i >= 0; i --)
		for(int j = i + 1; j < n; j ++)
		s[i][n] -= s[i][j] * s[j][n];//消成一元 
	
	return 0;
}
 
int main()
{
	cin >> n;
	for(int i = 0; i < n; i ++)
		for(int j = 0; j < n + 1; j ++)
			cin >> s[i][j];
			
	int d = gauss();
	
	if(d == 0)//有解 
	{
		for(int i = 0; i < n; i ++)
		{
			if(fabs(s[i][n]) < eps) s[i][n] = 0;
			printf("%.2lf\n", s[i][n]);
		} 
	}
	else if(d == 1)//无数解 
		puts("Infinite group solutions");
	else puts("No solution");
	
	return 0;
}