Cod sursa(job #2308873)

Utilizator HaesteinnSabau Florin Vlad Haesteinn Data 27 decembrie 2018 21:20:59
Problema Algoritmul lui Dijkstra Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.29 kb
#include <vector>
#include <fstream>
#include <queue>
#include <climits>

using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

vector<pair<int,int> > adj[50002];
int dist[50002];
bool visited[50002];

void dijkstra(int s)
{
    priority_queue<pair<int,int>,vector<pair<int,int> >, greater<pair<int,int> > > pq;
    pq.push({0,s});
    visited[s]=1;
    dist[s]=0;
    while(!pq.empty())
    {
        auto v=pq.top().second;
        pq.pop();
        for(auto u:adj[v])
        {
            //cout<<v<<" ";
            int weight=u.first;
            int vertex=u.second;
            if(dist[vertex]>dist[v]+weight)
            {
                dist[vertex]=dist[v]+weight;
                pq.push(u);
                //visited[vertex]=1;
            }
            //dist[vertex]=min(dist[vertex],dist[v]+weight);
        }
    }
}

int main()
{
    int n,m;
    fin>>n>>m;
    for (int i = 0; i < m; i++)
    {
        int v1,v2,w;
        fin>>v1>>v2>>w;
        adj[v1].push_back({w,v2});
    }
    for (int i = 0; i <= n; i++) {
        dist[i]=INT_MAX;
    }
    dijkstra(1);
    for(int i=2;i<=n;i++)
        if(dist[i]==INT_MAX)
            fout<<0<<" ";
        else
            fout<<dist[i]<<" ";
    return 0;
}