Cod sursa(job #1414040)

Utilizator retrogradLucian Bicsi retrograd Data 2 aprilie 2015 12:03:54
Problema Algoritmul lui Gauss Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.74 kb
#include<fstream>
#include<vector>
#include<cmath>
#include<iomanip>
#include<cstring>

using namespace std;
typedef int var;

ifstream fin("gauss.in");
ofstream fout("gauss.out");

#define MAXN 500
#define eps 1e-3
#define eq(a, b) (abs(a-b)<eps)

var m, n;
double A[MAXN][MAXN], B[MAXN][MAXN];
double X[MAXN];

void read() {
    fin>>n>>m;
    for(var i=1; i<=n; i++) {
        for(var j=1; j<=m+1; j++) {
            fin>>A[i][j];
            B[i][j] = A[i][j];
        }
    }
}

bool diag() {
    for(var step=1; step<=n; step++) {
        if(eq(B[step][step], 0.0)) {
            var i;
            for(i=step+1; i<=n && eq(B[i][step], 0); i++);

            if(i > n) continue;

            swap(B[i], B[step]);
        }

        for(var i=m+1; i>=step; i--) {
            B[step][i] /= B[step][step];
        }

        for(var i=step+1; i<=n; i++) {
            var inm = B[i][step];
            for(var j=step; j<=m+1; j++) {
                B[i][j] -= B[step][j] * inm;
            }
        }

    }

    return true;
}

void findX() {
    for(var i=n; i>=1; i--) {
        double rez = B[i][m+1];
        for(var j=m; j>i; j--) {
            rez -= B[i][j]*X[j];
        }
        X[i] = rez;
    }
}

int main() {

    read();
    if(!diag()) {
        fout<<"Imposibil";
        return 0;
    }

    findX();
/*
    for(var i=1; i<=n; i++) {
        double rez = 0.0;
        for(var j=1; j<=m; j++) {
            rez += A[i][j] * X[j];
        }
        if(!eq(rez, A[i][m+1])) {
            fout<<"Imposibil";
            return 0;
        }
    }
*/
    fout<<fixed<<setprecision(10);

    for(var i=1; i<=m; i++) {
        fout<<X[i]<<" ";
    }

    return 0;
}