Cod sursa(job #2184677)

Utilizator MaligMamaliga cu smantana Malig Data 24 martie 2018 10:38:32
Problema Algoritmul lui Gauss Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 2.1 kb
#include <bits/stdc++.h>

using namespace std;

#if 1
    #define pv(x) cout<<#x<<" = "<<x<<"; ";cout.flush()
    #define pn cout<<endl
#else
    #define pv(x)
    #define pn
#endif

#ifdef ONLINE_JUDGE
    #define in cin
    #define out cout
#else
    ifstream in("gauss.in");
    ofstream out("gauss.out");
#endif

using ll = long long;
using ull = unsigned long long;
using uint = unsigned int;
#define pb push_back
#define mp make_pair
const int NMax = 3e6 + 5;
const ll inf_ll = 1e18 + 5;
const int inf_int = 1e9 + 5;
const int mod = 100003;
const double eps = 1e-7;
using zint = int;

int main() {
    int N,M;
    in >> N >> M;

    vector< vector<double> > a(N+1);
    vector< double > X(N+1,0.0);
    for (int i = 1;i <= N;++i) {
        a[i] = vector<double> (M+2);

        for (int j = 1;j <= M+1;++j) {
            in >> a[i][j];
        }
    }

    int i = 1, j = 1;
    while (i <= N && j <= M) {

        int k;
        for (k = i;k <= N;++k) {
            if (a[k][j] < -eps || eps < a[k][j]) {
                break;
            }
        }

        if (k == N+1) {
            ++j;
            continue;
        }

        swap(a[i],a[k]);

        for (int u = j+1;u <= M+1;++u) {
            a[i][u] /= a[i][j];
        }
        a[i][j] = 1;

        for (int k = i+1;k <= N;++k) {
            for (int u = j+1;u <= M+1;++u) {
                a[k][u] -= a[k][j] * a[i][u];
            }
            a[k][j] = 0;
        }

        ++i;++j;
    }

    for (int i = N;i > 0;--i) {
        int j;
        for (j = 1;j <= M+1;++j) {
            if (a[i][j] < -eps || eps < a[i][j]) {
                break;
            }
        }

        if (j == M+2) {
            continue;
        }
        if (j == M+1) {
            out << "Imposibil\n";
            return 0;
        }

        X[j] = a[i][M+1];
        for (int k = j+1;k <= M;++k) {
            X[j] -= a[i][k] * X[k];
        }
    }

    fixed(out);
    out.precision(8);
    for (int i = 1;i <= N;++i) {
        out << X[i] << ' ';
    }

    return 0;
}