【高斯消元】【模板】高斯消元法

链接

$luoguP3389$

题目描述

给定一个线性方程组，对其求解

输入&输出

第一行，一个正整数 $n$

第二至$n+1$行，每行 $n+1$ 个整数，为$a_1,a_2\cdots a_n$和 $b$，代表一组方程。

共$n$行，每行一个数，第 $i$行为 $x_i$（保留2位小数）

如果不存在唯一解，在第一行输出"No Solution".

样例输入

3
1 3 4 5
1 4 7 3
9 3 2 2

样例输出

-0.97
5.18
-2.39

数据范围或提示

$1\le n\le 100,\mid a_i\mid \le 10^4,\mid b\mid \le 10^4$

思路

不了解高斯消元法的建议先自行查阅一下：传送门
每次找到每列系数最大的，然后换到当前行，再与下面每一行都进行消元

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#define Abs(x) (x) < 0 ? -(x) : (x)

using namespace std;

int n;
double a[101][102];

int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n + 1; j++)
			scanf("%lf", &a[i][j]);
	
	for (int i = 1; i <= n; i++) {
		int maxx = i;
		for (int j = i + 1; j <= n; j++)
			if (Abs(a[j][i]) > Abs(a[maxx][i]))
				maxx = j;
		for (int j = 1; j <= n + 1; j++)
			swap(a[i][j], a[maxx][j]);
		if (!a[i][i]) {
			printf("No Solution");
			return 0;
		}
		for (int j = 1; j <= n; j++)
			if (i != j) {
				for (int k = i + 1; k <= n + 1; k++)
					a[j][k] = a[j][k] - (a[i][k] * (a[j][i] / a[i][i]));
			}
	}
	
	for (int i = 1; i <= n; i++)
		printf("%.2lf\n", a[i][n + 1] / a[i][i]);
	
	return 0;
}