Cod sursa(job #3215644)

Utilizator dobreraduDobre Radu Fabian dobreradu Data 15 martie 2024 11:10:11
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
const int NMAX = 5e4 + 5;
const int INF = 2e9;
struct Radu{
    int cost, next;
};
struct HeapNode{
    int cost, next;
    bool operator < ( const HeapNode &other ) const{
        return other.cost < cost;
    }
};
int n, dist[NMAX];
bool viz[NMAX];
vector <Radu> g[NMAX];
priority_queue <HeapNode> q;
void dijkstra( int node ){
    for( int i = 2; i <= n; i++ )
        dist[i] = INF;
    q.push({0, node});
    while( !q.empty() ){
        node = q.top().next;
        q.pop();
        for( auto vecin : g[node] ){
            if( vecin.cost + dist[node] < dist[vecin.next] ){
                dist[vecin.next] = vecin.cost + dist[node];
                q.push({dist[vecin.next], vecin.next});
            }
        }
    }
}
int main()
{
    int n, m;
    in >> n >> m;
    for( int i = 0; i < m; i++ ){
        int a, b, c;
        in >> a >> b >> c;
        g[a].push_back({c, b});
    }
    dijkstra(1);
    for( int i = 2; i <= n; i++ )
        if( dist[i] == INF )
            out << 0 << " ";
        else
            out << dist[i] << " ";

    return 0;
}