Cod sursa(job #2539992)

Utilizator emanuel2186Lugojan Emanuel emanuel2186 Data 6 februarie 2020 17:00:00
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.54 kb
#include <bits/stdc++.h>
#define NMAX 50005
using namespace std;
ifstream fin ("dij.in");
ofstream fout ("dij.out");
int n,m;
int a,b,c;
bool viz[NMAX];
int D[NMAX];

deque<pair <int , int > > vec[NMAX];

struct compar
{
    bool operator()(int x, int y)
    {
        return D[x] > D[y];
    }
};

priority_queue<int , vector<int> , compar > coada;

void dij(int start)
{
    for (int i=1; i<=n; i++)
        D[i] = INT_MAX;
    D[start] = 0;

    coada.push(start);
    viz[start] = true;

    while(!coada.empty())
    {
        int nod_curent = coada.top();

        coada.pop();
        viz[nod_curent] = 0;

        for (int i=0; i<vec[nod_curent].size(); i++)
        {
            int vecin = vec[nod_curent][i].first;
            int cost = vec[nod_curent][i].second;

            if (D[vecin] > D[nod_curent] + cost)
            {
                D[vecin] = D[nod_curent] + cost;

                if (viz[vecin] == 0)
                {
                    coada.push(vecin);
                    viz[vecin] = 1;
                }
            }
        }
    }
}
void afisare()
{
    for (int i=2; i<=n; i++)
    {
        if (D[i] == INT_MAX)
            fout<<0<<" ";
        else
            fout<<D[i]<<" ";
    }
    fout<<"\n";
    fin.close();
    fout.close();
}
void citire()
{
    fin>>n>>m;
    for (int i=1; i<=m; i++)
    {
        fin>>a>>b>>c;
        vec[a].push_back(make_pair(b,c));
    }

    dij(1);
    afisare();

}
int main()
{
    citire();
    return 0;
}