Cod sursa(job #1811466)

Utilizator triscacezarTrisca Vicol Cezar triscacezar Data 21 noiembrie 2016 11:40:52
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.88 kb
#include <fstream>
#include <vector>
#include <set>
#define INF 20010
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
vector<pair<int,int> > v[50010];
set<int> s;
int n,m,x,i,y,w,alt,dist[50010];
int main()
{
    f>>n>>m;
    for(;m;m--)
    {
        f>>x>>y>>w;
        v[x].push_back(make_pair(y,w));
    }
    for(i=1;i<=n;i++)
    {
        dist[i]=INF;
        s.insert(i);
    }
    dist[1]=0;
    while(s.size())
    {
        x=*s.begin();
        for(auto it:s)
            if(dist[it]<dist[x])
                x=it;
        s.erase(x);
        for(auto it:v[x])
        {
            alt=it.second+dist[x];
            if(alt<dist[it.first])
                dist[it.first]=alt;
        }
    }
    for(i=2;i<=n;i++)
        if(dist[i]!=INF)
            g<<dist[i]<<' ';
        else
            g<<0;
    return 0;
}