Cod sursa(job #1479061)

Utilizator tudormaximTudor Maxim tudormaxim Data 30 august 2015 14:49:27
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <bits/stdc++.h>

using namespace std;
const int nmax = 50005;
int n, m, dist[nmax];
vector <pair<int,int> > graf[nmax];

void dijkstra()
{
    set <pair<int,int> > s;
    int i, d, nod;
    for(i=2; i<=n; i++)
        dist[i]=INFINITY;

    s.insert(make_pair(1, 0));
    while(!s.empty())
    {
        nod=s.begin()->first;
        d=s.begin()->second;
        s.erase(s.begin());
        for(i=0; i<graf[nod].size(); i++)
            if(dist[graf[nod][i].first] > d+graf[nod][i].second)
            {
                dist[graf[nod][i].first]=d+graf[nod][i].second;
                s.insert(make_pair(graf[nod][i].first, dist[graf[nod][i].first]));
            }
    }
}

int main()
{
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);
    int x, y, c, i;
    scanf("%d %d", &n, &m);
    for(i=1; i<=m; i++)
    {
        scanf("%d %d %d", &x, &y, &c);
        graf[x].push_back(make_pair(y, c));
    }
    dijkstra();
    for(i=2; i<=n; i++)
    {
        if(dist[i]==INFINITY) dist[i]=0;
        printf("%d ", dist[i]);
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}