Cod sursa(job #3213825)

Utilizator raulababeiAbabei Raul raulababei Data 13 martie 2024 15:01:49
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <bits/stdc++.h>
#define NMAX 10001
#define INF 1e9

using namespace std;

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

vector<pair<int, int>> vec[NMAX * 5];
int dp[NMAX * 5];
priority_queue<pair<int, int>> q;
int viz[NMAX * 5];

void dijkstra(int x){
    q.push({0, x});
    dp[x] = 0;
    while(!q.empty()){
        int nod = q.top().second;
        q.pop();
        if(viz[nod] == 0){
            viz[nod] = 1;
            for(auto w : vec[nod]){
                if(!viz[w.second]){
                    if(dp[nod] + w.first < dp[w.second]){
                        dp[w.second] = dp[nod] + w.first;
                        q.push({-dp[w.second], w.second});
                    }
                }
            }
        }
    }
}

int main() {



    int n;
    int m;
    in >> n >> m;
    for(int i = 0;i < m;i++){
        int a, b, cost;
        in >> a >> b >> cost;
        vec[a].push_back({cost, b});
    }

    for(int i = 0;i <= n;i++){
        dp[i] = INF;
    }

    dijkstra(1);

    for(int i = 2;i <= n;i++){
        if(dp[i] == INF){
            out << "0 ";
        }
        else out << dp[i] << ' ';
    }

    return 0;
}