Cod sursa(job #2345395)

Utilizator radumihaisirbuSirbu Radu-Mihai radumihaisirbu Data 16 februarie 2019 12:18:32
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.24 kb
#include <bits/stdc++.h>

using namespace std;

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

const int INF = 2e9;

struct muchie
{
    int Y, cost;
}A;

int n, m, x, y, C;

int NOD;

vector <muchie> v[50005];
vector <int> dist(50005, INF);

void dijkstra (int nod)
{

    bitset <50005> viz;
    dist[nod] = 0;
    priority_queue< pair <int, int>, vector< pair <int, int> >, greater <pair <int , int> > > q;
    q.push(make_pair(0, nod));
    while(!q.empty())
    {
        NOD = q.top().second;
        viz[NOD] = 1;
        q.pop();
        for (vector <muchie> :: iterator i=v[NOD].begin(); i!=v[NOD].end(); i++)
        {
            A = *i;
            if (dist[NOD] + A.cost < dist[A.Y])
            {
                dist[A.Y] = dist[NOD] + A.cost;
                if (viz[A.Y] == 0)
                {
                    q.push(make_pair(dist[A.Y], A.Y));
                }
            }
        }
    }
}

int main()
{
    fin >> n >> m;
    for (int i=0; i<m; i++)
    {
        fin >> x >> A.Y >> A.cost;
        v[x].push_back(A);
    }
    dijkstra(1);
    for (int i=2; i<=n; i++)
        if (dist[i] == INF) fout << "0 ";
        else fout << dist[i] << " ";
    return 0;
}