Cod sursa(job #2331067)

Utilizator albucristianAlbu Cristian-Gabriel albucristian Data 29 ianuarie 2019 09:54:03
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 kb
#include <fstream>
#include <vector>
#include <bitset>
#include <queue>
#define nmax 50002
#define inf 1000000000
using namespace std;
class cmp
{
public:
    bool operator()(pair<int,int>a,pair<int,int>b)
    {
        return a.second>b.second;
    }
};
int dist[nmax];
priority_queue <pair<int,int>,vector<pair<int,int>>,cmp>coada;
vector<pair<int,int>>g[nmax];
bitset <nmax> vizitat;
int main()
{
    int n,m,a,b,nod,x,y,c;
    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++)
            {
                x=g[nod][i].first;
                y=g[nod][i].second;
                if(dist[x]>dist[nod]+y)
                {
                    dist[x]=dist[nod]+y;
                    coada.push(make_pair(x,dist[x]));
                }
            }
            vizitat[nod]=1;
        }
    }
    for(int i=2;i<=n;i++)
    {
        if(dist[i]==inf)
            dist[i]=0;
        out<<dist[i]<<" ";
    }
    return 0;
}