Cod sursa(job #2159280)

Utilizator antracodsAntracod antracods Data 10 martie 2018 20:51:21
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.34 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>

using namespace std;

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

const int NMAX = 50003;
const int MMAX = 250003;
const int INF = 1 << 30;

int used[NMAX];
int sol[NMAX];
int n,m;

vector <pair<int,int> > Graf[NMAX];
priority_queue <pair<int,int>> my_queue;

void citire()
{
    in>>n>>m;
    for(int i=1; i<=m; i++)
    {
        int x1,y1,z1;
        in>>x1>>y1>>z1;
        Graf[x1].push_back({y1,z1});
    }
    for(int i=2; i<=n; i++)
    {
        sol[i]=INF;
    }

}

void dijkstra()
{
    my_queue.push(make_pair(0,1));
    while(my_queue.empty()==0)
    {
        int node=my_queue.top().second;
        my_queue.pop();
        if(used[node]==1)
            continue;
        used[node]==1;
        for(register int i=0; i<Graf[node].size(); i++)
        {
            if(sol[Graf[node][i].first]>Graf[node][i].second+sol[node])
            {
                sol[Graf[node][i].first]=Graf[node][i].second+sol[node];
                my_queue.push({-sol[Graf[node][i].first],Graf[node][i].first});
            }
        }
    }
}

int main()
{
    citire();
    dijkstra();
    for(int i=2; i<=n; i++)
    {
        if(sol[i]==INF)
            out<<0<<" ";
        else
            out<<sol[i]<<" ";
    }
}