Cod sursa(job #3227595)

Utilizator CXHRISXhamani Christian CXHRIS Data 2 mai 2024 09:17:00
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <bits/stdc++.h>
#define INF 1000000000
using namespace std;

int n, i, j, k, c, p, pmax, a[105][105], v[105], d[105], t[105];
int main()
{
    ifstream f("dijkstra.in");
    ofstream g("dijkstra.out");
    f >> n >> p;
    for(i = 1; i <= n; i ++)
     for(j = 1; j <= n; j ++) a[i][j] = INF;
    while(f >> i >> j >> c) a[i][j] = c;
    for(i = 1; i <= n; i ++)
    {
        v[i] = 0;
        d[i] = a[p][i];
        t[i] = p;
    }
    v[p] = 1;
    t[p] = 0;
    d[p] = 0;
    d[0] = INF;
    for(k = 1; k < n; k ++)
    {
        pmax = 0;
        for(i = 1; i <= n; i ++)
         if(v[i] == 0 && d[i] < d[pmax]) pmax = i;
        if(pmax > -1)
        {
            v[pmax] = 1;
            for(i = 1; i <= n; i ++)
             if(v[i] == 0 && d[i] > d[pmax] + a[pmax][i])
              d[i] = d[pmax] + a[pmax][i], t[i] = pmax;
        }
    }
    for(i = 1; i <= n; i ++)
     g << (d[i] != INF ? d[i] : -1) << " ";
    return 0;
}