Cod sursa(job #3264326)

Utilizator burcuriciBucur Stefan burcurici Data 20 decembrie 2024 13:46:08
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.25 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

int n, m;
bool f[50005];
vector<int> d;
vector<pair<int,int>> a[50005];
priority_queue<pair<int,int> , vector<pair<int,int>>, greater<pair<int,int>>> Q;

void citire()
{
    fin>>n>>m;
    d.resize(50005,1e5);
    for(int i=1; i<=m; i++)
    {
        int x,y,s;
        fin>>x>>y>>s;
        if(x==1) d[y]=s;
        a[x].push_back(make_pair(s,y));
    }
    d[1]=0;
    f[1]=1;
    for(auto i:a[1]){
        Q.push(i);
        d[i.second]=i.first;
    }
}

void ex()
{
    while(!Q.empty()){
        while(f[Q.top().second]==1 && !Q.empty())
                Q.pop();
        if(Q.empty()) continue;
        auto x=Q.top();
        Q.pop();
        f[x.second]=1;
        for(auto i:a[x.second]){
            if(d[i.second]>d[x.second]+i.first)
                d[i.second] = d[x.second]+i.first;
            if(f[i.second]==0) Q.push(make_pair(d[i.second],i.second));
        }
    }
}

void afisare()
{
    for(int i=2; i<=n; i++){
        if(d[i]==1e5) fout<<"0 ";
        else fout<<d[i]<<" ";
    }
}

int main()
{
    citire();
    ex();
    afisare();
}