Cod sursa(job #2071199)

Utilizator Y.MalmsteenB.P.M. Y.Malmsteen Data 20 noiembrie 2017 14:31:11
Problema Algoritmul lui Gauss Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 2.41 kb
#include <iostream>
#include <fstream>
#include <iomanip>
#include <cmath>
using namespace std;

const double EPS = 1e-8;
const int MAXN = 301;
ifstream f("gauss.in");
ofstream g("gauss.out");

int N, M;
double A[MAXN][MAXN];
double X[MAXN];

void gauss()
{
    int i = 1, j = 1, k;
    while(i <= N && j <= M)
    {
        //Cautam o linie k pentru care A[k][j] != 0
        bool ok = 0;
        for(k = i; k <= N; ++k)
            if(abs(A[k][j]) > EPS) //A[k][j] != 0
            {
                ok = 1;
                break;
            }
        //Daca nu gasim ==> necunoscuta j este variabila libera,
        //incrementam pe j (trecem la coloana urmatoare)
        if(ok == 0)//if(k == N + 1)
        {
            j++;
            continue;
        }
        //Interschimbam linia k cu linia i, daca k != i
        if(k != i)
            for(int l = 1; l <= M + 1; l++)
                swap(A[i][l], A[k][l]);
        //Impartim toate valorile de pe linia i la A[i][j], A[i][j] devenind 1.
        //(Valorile de pe linia i si coloanele 1..j-1 sunt deja 0)
        for(int l = j + 1; l <= M + 1; l++)
            A[i][l] = A[i][l] / A[i][j];
        A[i][j] = 1;
        //Scadem din ecuatiile i+1..N ecuatia i inmultita cu A[u][j]
        //pentru a egala toti coeficientii de coloana a acestor linii la 0.
        for(int u = i + 1; u <= N; u++)
        {
            for(int l = j + 1; l <= M + 1; l++)
                A[u][l] -= A[u][j] * A[i][l];
            A[u][j] = 0;
        }
        i++;
        j++;
    }
}

bool solutie()
{
    for(int i = N; i >= 1; i--)
        for(int j = 1; j <= M + 1; j++)
            if(abs(A[i][j]) > EPS) // A[i][j] != 0
            {
                if(j == M + 1)
                    return 0; //Sistemul este incompatibil
                X[j] = A[i][M + 1]; //Calculam pe necunoscuta j
                for(int k = j + 1; k <= M; k++)
                    X[j] -= X[k] * A[i][k];
                break;
            }
    return 1;
}

void afisare()
{
    g.setf(ios::fixed, ios::floatfield);
    g.precision(4);
    for(int i = 1; i <= M; i++)
        g << X[i] << ' ';
}

int main()
{
    f >> N >> M;
    for(int i = 1; i <= N; ++i)
        for(int j = 1; j <= M + 1; ++j)
            f >> A[i][j];
    //
    gauss();
    //
    if(solutie())
        afisare();
    else
        g << "Imposibil";
    return 0;
}