Cod sursa(job #2331647)

Utilizator albucristianAlbu Cristian-Gabriel albucristian Data 29 ianuarie 2019 19:24:00
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#define nmax 50002
#define inf 100000000
using namespace std;
class cmp{
public:
    bool operator()(pair<int,int>a,pair<int,int>b)
    {
        return a.second>b.second;
    }
};
priority_queue<pair<int,int>,vector<pair<int,int>>,cmp>coada;
vector<pair<int,int>>g[nmax];
bitset <nmax> vizitat;
int dist[nmax];
int main()
{
    int n,m,a,b,c,nod;
    ifstream in("dijkstra.in");
    ofstream out("dijkstra.out");
    in>>n>>m;
    for(int i=1;i<=m;i++)
    {
        in>>a>>b>>c;
        g[a].push_back(make_pair(b,c));
    }
    for(int i=1;i<=n;i++)
    {
        dist[i]=inf;
    }
    dist[1]=0;
    coada.push(make_pair(1,0));
    while(!coada.empty())
    {
        nod=coada.top().first;
        coada.pop();
        if(!vizitat[nod])
        {
            for(int i=0;i<g[nod].size();i++)
            {
                a=g[nod][i].first;
                b=g[nod][i].second;
                if(dist[a]>dist[nod]+b)
                {
                    dist[a]=dist[nod]+b;
                    coada.push(make_pair(a,dist[a]));
                }
            }
            vizitat[nod]=1;
        }
    }
    for(int i=2;i<=n;i++)
    {
        if(dist[i]==inf)
            dist[i]=0;
        out<<dist[i]<<" ";
    }
    return 0;
}