Cod sursa(job #2143654)

Utilizator ana_dariaAna Daria Hendoreanu ana_daria Data 26 februarie 2018 10:09:16
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.18 kb
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;
typedef pair<int,int> ii;
const int inf=100000000,nmax=50000;
vector <ii> G[nmax+5];
bool viz[nmax+3];
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    int n,m, u,v,w;
    scanf("%d%d",&n,&m);
    ii aux;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        aux.first=w;
        aux.second=v;
        G[u].push_back(aux);
    }
    vector<int> dist(n,inf);
    dist[1]=0;
    priority_queue<ii, vector<ii>, greater<ii> > pq;
    pq.push(ii(0,1));
    ii asdf;
    while(!pq.empty())
    {
        asdf=pq.top();
        pq.pop();
        int d=asdf.first,u=asdf.second;
        if(d>dist[u])
            continue;
        for(int j=0;j<(int)G[u].size();j++)
        {
            int v=G[u][j].second,duv=G[u][j].first;
            viz[v]=true;
            if(dist[u]+duv<dist[v])
            {
                dist[v]=dist[u]+duv;
                pq.push(ii(dist[v],v));
            }
        }
    }
    for(int i=2;i<=n;i++)
        if(!viz[i]) printf("0 ");
        else printf("%d ",dist[i]);
    return 0;
}