Cod sursa(job #2236289)

Utilizator parsulPaul Cristian Banu-Taran parsul Data 29 august 2018 00:31:23
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
#include<fstream>
#include<vector>
#include<queue>
#define mp make_pair
#define pb push_back
#define NMAX 50009
#define MAX (1<<30)
using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
int n,m,A,B,nc,d[NMAX];
short C;
bool viz[NMAX];
vector<pair<int,short> >v[NMAX];
struct minimum
{
    bool operator()(int A, int B)
    {
        return d[A] > d[B];
    }
};
priority_queue<int,vector<int>,minimum> q;
void citire()
{
    in>>n>>m;
    for(int i=0;i<m;i++)
    {
        in>>A>>B>>C;
        v[A].pb(mp(B,C));
    }
}
void prelucrare()
{
    for(int i=2;i<=n;i++)
        d[i]=MAX;
    d[1]=0;
    q.push(1);
    viz[1]=1;
    while(!q.empty())
    {
        nc=q.top();
        q.pop();
        viz[nc]=0;
        for(unsigned i=0;i<v[nc].size();i++)
            if(d[nc]+v[nc][i].second<d[v[nc][i].first])
        {
            d[v[nc][i].first]=d[nc]+v[nc][i].second;
            if(!viz[v[nc][i].first])
            {
                q.push(v[nc][i].first);
                viz[v[nc][i].first]=1;
            }
        }
    }

}
void tipar()
{
    for(int i=2;i<=n;i++)
        if(d[i]==MAX)
            out<<0<<" ";
        else
            out<<d[i]<<" ";
}
int main()
{
    citire();
    prelucrare();
    tipar();
}