洛谷-3389 【模板】高斯消元法

题目描述
给定一个线性方程组，对其求解
输入格式
第一行，一个正整数 n
第二至 n+1行，每行 n+1 个整数，为a1,a2⋯an和 b，代表一组方程。
输出格式
共n行，每行一个数，第 i行为xi （保留2位小数）
如果不存在唯一解，在第一行输出"No Solution".

输入输出样例
输入 #1
3
1 3 4 5
1 4 7 3
9 3 2 2

输出 #1
-0.97
5.18
-2.39

说明/提示
1≤n≤100,∣ai∣≤10⁴,∣b∣≤10⁴

#include<iostream>
#include<cstdio>
#include<cmath>
#define N 105
using namespace std;
double a[N][N]={0};
double X[N]={0};
int n=0;
bool guess(){
    for(int i=1;i<=n;i++){
        int max_index=i;
        for(int j=i;j<=n;j++) if(a[j][i]>a[max_index][i]) max_index=j;
        if(fabs(a[max_index][i])<1e-5) return 0;
        for(int j=i;j<=n+1;j++) swap(a[i][j],a[max_index][j]);
        for(int j=1;j<=n;j++) if(j!=i) for(int k=n+1;k>=i;k--) a[j][k]-=a[j][i]/a[i][i]*a[i][k];
    }
    for(int i=1;i<=n;i++) X[i]=a[i][n+1]/a[i][i];
    return 1;
}
int main(){
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1;i<=n;i++) for(int j=1;j<=n+1;j++) cin>>a[i][j];
    if(!guess()) cout<<"No Solution"<<endl;
    else{
        for(int i=1;i<=n;i++) printf("%.2f\n",X[i]);
    }
    return 0;
}