Cod sursa(job #1335642)

Utilizator bogdanboboc97Bogdan Boboc bogdanboboc97 Data 5 februarie 2015 19:39:48
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.84 kb
#include <bits/stdc++.h>
#define INF numeric_limits<int>::max()
using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
vector< vector< pair<int,int> > > a;
priority_queue< pair<int,int> > q;
vector<int> d;
void dijkstra(int x)
{
    q.push({0,x});
    d[x]=0;
    while(!q.empty())
    {
        int x=q.top().second;
        q.pop();
        for(auto i: a[x])
        if(d[x] + i.second < d[i.first])
        {
            d[i.first]=d[x]+i.second;
            q.push({-d[i.first],i.first});
        }
    }
}
int main()
{
    int n,m,x,y,z;
    in>>n>>m;
    a=vector< vector< pair<int,int> > >(n+1);
    d=vector<int>(n+1,INF);
    for(;m;m--)
    {
        in>>x>>y>>z;
        a[x].push_back({y,z});
    }
    dijkstra(1);
    for(int i=2;i<=n;i++)out<<(d[i]==INF ? 0 : d[i])<<' ';
    return 0;
}