Cod sursa(job #1293207)

Utilizator Al3ks1002Alex Cociorva Al3ks1002 Data 15 decembrie 2014 16:21:44
Problema Algoritmul lui Gauss Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.9 kb
#include<cstdio>
#include<fstream>
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<vector>
#include<bitset>
#include<deque>
#include<queue>
#include<set>
#include<map>
#include<cmath>
#include<cstring>
#include<ctime>
#include<cstdlib>
#include<unordered_map>

#define ll long long
#define pb push_back
#define mp make_pair
#define pii pair<int,int>
#define pll pair<ll,ll>

using namespace std;

const int nmax = 305;
const double eps = 1e-10;

int n, m, i, j, k, l;
double a[nmax][nmax];
double sol[nmax];

bool dif(double x)
{
    return fabs(x) > eps;
}

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

    fin >> n >> m;

    for(i = 1; i <= n; i++)
        for(j = 1; j <= m + 1; j++)
            fin >> a[i][j];

    for(i = 1, j = 1; i <= n && j <= m;)
    {
        for(k = i; k <= n; k++)
            if(dif(a[k][j]))
                break;

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

        if(i != k)
        {
            for(l = j; l <= m + 1; l++)
                swap(a[i][l], a[k][l]);
        }

        for(l = j + 1; l <= m + 1; l++)
            a[i][l] /= a[i][j];

        a[i][j] = 1.0;

        for(k = i + 1; k <= n; k++)
        {
            for(l = j + 1; l <= m + 1; l++)
                a[k][l] -= a[k][j] * a[i][l];

            a[k][j] = 0.0;
        }

        i++;
        j++;
    }

    for(i = n; i; i--)
    {
        for(j = 1; j <= m + 1; j++)
            if(dif(a[i][j]))
                break;

        if(j == m + 1)
        {
            fout << "Imposibil\n";
            return 0;
        }

        sol[j] = a[i][m + 1];
        for(k = j + 1; k <= m; k++)
            sol[j] -= sol[k] * a[i][k];
    }

    for(i = 1; i <= m; i++)
        fout << fixed << setprecision(11) << sol[i] << " ";

    return 0;
}