Cod sursa(job #2664333)

Utilizator metallidethantralayerIon Cojocaru metallidethantralayer Data 28 octombrie 2020 15:23:09
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("dijsktra.in");
ofstream g("dijkstra.out");
int n,m,D[50005];
struct Compare
{
    bool operator ()(const int &a,const int &b)
    {
        return D[a]>D[b];
    }
};
vector <pair<int,int> > V[50005];
priority_queue <int,vector <int>, Compare> Q;
bitset <50005> InCoada;
int main()
{
    f>>n>>m;
    for(int i=1; i<=m; i++)
    {
        int x,y,z;
        f>>x>>y>>z;
        V[x].push_back({y,z});
    }
    for(int i=2; i<=n; i++)
        D[i]=1e9;
    Q.push(1);
    InCoada[1]=true;
    while(!Q.empty())
    {
        int x=Q.top();
        Q.pop();
        InCoada[x]=false;
        for(int i=0; i<V[x].size(); i++)
            if(D[V[x][i].first]>D[x]+V[x][i].second)
            {
                D[V[x][i].first]=D[x]+V[x][i].second;
                if(!InCoada[V[x][i].first])
                {
                    InCoada[V[x][i].first]=true;
                    Q.push(V[x][i].first);
                }
            }
    }
    for(int i=2; i<=n; i++)
        if(D[i]==1e9)
            g<<0<<' ';
        else
            g<<D[i]<<' ';

    return 0;
}