Cod sursa(job #2923469)

Utilizator AndreiBOTOBotocan Andrei AndreiBOTO Data 14 septembrie 2022 15:42:30
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <queue>
#include <vector>

///#include <tryhardmode>
///#include <GODMODE::ON>

using namespace std;

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

const int NMAX=5e4+5;
const int NMAX2=2e9;

int dist[NMAX];
int viz[NMAX];

priority_queue<pair<int,int>>q;
vector<pair<int,int>>v[NMAX];

void dijkstra(int p)
{
    dist[p]=0;
    viz[p]=1;
    q.push(make_pair(0,p));
    while(!q.empty())
    {
        p=q.top().second;
        q.pop();
        viz[p]=0;
        for(auto i:v[p])
        {
            if(dist[p]+i.second<dist[i.first])
            {
                dist[i.first]=dist[p]+i.second;
                if(viz[i.first]==0)
                {
                    viz[i.first]=1;
                    q.push(make_pair(-dist[i.first],i.first));
                }
            }
        }
    }
}

int main()
{
    int n,m,i,j,a,b,c;
    fin>>n>>m;
    while(m--)
    {
        fin>>a>>b>>c;
        v[a].push_back(make_pair(b,c));
    }
    for(i=1;i<=n;i++)
        dist[i]=NMAX2;
    dijkstra(1);
    for(i=1;i<=n;i++)
        if(i!=1)
    {
        if(dist[i]==NMAX2)
            dist[i]=0;
        fout<<dist[i]<<" ";
    }
    return 0;
}