Cod sursa(job #3163566)

Utilizator ionut3007Malita Ionut ionut3007 Data 31 octombrie 2023 17:12:29
Problema Algoritmul lui Dijkstra Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.33 kb
#include <bits/stdc++.h>

using namespace std;

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

const int NMax=50005;
const int MMax=250005;
const int oo=(1<<30);

int n,m;
int d[NMax];
bool sw[NMax];

vector<pair<int,int>>v[MMax];

struct comp{
    bool operator()(int x,int y){
        return d[x]>d[y];
    }
};
priority_queue<int,vector<int>,comp>Q;

void cit(){
    fin>>n>>m;
    for(int i=1;i<=m;i++){
        int x,y,c;
        fin>>x>>y>>c;
        v[x].push_back(make_pair(y,c));
    }
}

void dijkstra(int nods){
    for(int i=1;i<=n;i++)
        d[i]=oo;

    d[nods]=0;
    Q.push(nods);
    sw[nods]=true;
    while(!Q.empty()){
        int nodc=Q.top();
        sw[nodc]=true;
        for(size_t i=0;i<v[nodc].size();i++){
            int vecin=v[nodc][i].first;
            int cost=v[nodc][i].second;
            if(d[nodc]+cost<d[vecin]){
                d[vecin]=d[nodc]+cost;
                if(sw[vecin]==false){
                    sw[vecin]=true;
                    Q.push(vecin);
                }
            }
        }
        Q.pop();
    }
}

void afis(){
    for(int i=2;i<=n;i++)
        if(d[i]!=oo)
            fout<<d[i]<< " ";
        else
            fout<< 0<< " ";
}

int main()
{
    cit();
    dijkstra(1);
    afis();
    return 0;
}