Cod sursa(job #2974399)

Utilizator sandry24Grosu Alexandru sandry24 Data 4 februarie 2023 00:23:29
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.36 kb
#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pi;
#define pb push_back
#define mp make_pair
#define f first
#define s second

struct Node {
    int nr, dist;
    struct Node *next;
};

int n, m, parent[50001], dist[50001];
Node *lda[50001];
Node *r, *p, *v, *mine;

void add_edge(Node *&y, int x, int d){
    r = new Node;
    r->nr = x;
    r->dist = d;
    r->next = y;
    y = r;
}

Node* min_element(Node *h){
    Node *mine = h;
    while(h){
        if(mine->nr > h->nr)
            mine = h;
        h = h->next;
    }
    return mine;
}

void remove_element(Node *h){
    r = p; v = p;
    while(r){
        if(r != h){
            v = r;
            r = r->next;
        } else {
            if(r == p){
                p = r->next;
                delete r;
                v = p;
                r = p;
            } else if(r->next == NULL){
                v->next = NULL;
                delete r;
                r = NULL;
            } else {
                v->next = r->next;
                delete r;
                r = v->next;
            }
        }
    }
}

void Dijkstra(int x){
    dist[x] = 0;
    parent[x] = -1;
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q; //pentru a sorta crescator minheap
    q.push({0, x});
    p = r; v = r;
    while(!q.empty()){
        int a = q.top().second;
        q.pop();
        for(Node *h = lda[a]; h != NULL; h = h->next){
            int b = h->nr, d = h->dist;
            if(dist[a] + d < dist[b]){
                dist[b] = dist[a] + d;
                q.push({dist[b], b});
                parent[b] = a;
            }
        }
    }
}

void solve(){
    cin >> n >> m;
    for(int i = 0; i < m; i++){
        int a, b, d;
        cin >> a >> b >> d;
        add_edge(lda[a], b, d);
        //add_edge(lda[b], a, d); //daca graful este neorientat
    }  
    for(int i = 0; i <= 50000; i++)
        dist[i] = 1e9;
    Dijkstra(1);
    for(int i = 2; i <= n; i++)
        cout << (dist[i] == 1e9 ? 0 : dist[i]) << ' ';
}  
 
int main(){
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);
    ios::sync_with_stdio(0); cin.tie(0);
    int t = 1;
    //cin >> t;
    while(t--){
        solve();
    }
}