Cod sursa(job #1624301)

Utilizator AdrianVrAdrian Stefan Vramulet AdrianVr Data 2 martie 2016 10:09:41
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.1 kb
#include <bits/stdc++.h>
#define Nmax 50010
#define inf INT_MAX
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
struct arc
{
    int d,c;
};
vector <arc> M[Nmax];
priority_queue < int, vector <int> , greater <int> > heap;
int n,m,x,y,z,D[Nmax];
bool viz[Nmax];
void dijkstra(int R)
{
    int k;
    for(int i=1; i<=n; i++)
    {
        viz[i]=0;
        D[i]=inf/2;
    }
    D[R]=0;
    heap.push(R);
    while(heap.empty()==0)
    {
        k=heap.top();
        heap.pop();
        if(viz[k])
            continue;
        viz[k]=1;
        for(int i=0; i<M[k].size(); i++)
            if(viz[M[k][i].d]==0&& D[k]+M[k][i].c<D[M[k][i].d])
            {
                D[M[k][i].d]=D[k]+M[k][i].c;
                heap.push(M[k][i].d);
            }
    }
}
int main()
{
    f>>n>>m;
    arc aux;
    for(int i=1; i<=m; i++)
    {
        f>>x>>aux.d>>aux.c;

        M[x].push_back(aux);
    }
    dijkstra(1);
    for(int i=2; i<=n; i++)
        if(D[i]!=inf/2)
            g<<D[i]<<' ';
        else
            g<<0<<' ';
    return 0;
}