Cod sursa(job #2553270)

Utilizator salagean_brianaSalagean Briana salagean_briana Data 21 februarie 2020 20:10:25
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.37 kb
#include <iostream>
#include<fstream>
#include<queue>
#include<vector>
using namespace std;
#define pinf 1000000007
ifstream fin ("dijkstra.in");
ofstream fout ("dijkstra.out");
int n, m, viz[50010], minn;
unsigned long long d[50010];
priority_queue<pair<int,int>,vector<pair<int,int> >, std::greater<pair<int,int> > > q;
vector<pair<int,int> >v[50010];
void citire ()
{
    int x, y, co, i, j;
    fin>>n>>m;
    for (i=1; i<=m; i++)
    {
        fin>>x>>y>>co;
        v[x].push_back({co,y});
    }
    for (i=2; i<=n; i++)
        d[i]=pinf;
    for(i=0; i<v[1].size(); i++)
    {
        d[v[1][i].second]=v[1][i].first;
        q.push({v[1][i].first,v[1][i].second});
    }
}
void dijkstra ()
{
    int i, j;
    viz[1]=1;
    while (!q.empty())
    {
       minn=q.top().second;
       q.pop();
       viz[minn]=1;
       for (j=0; j<v[minn].size(); j++)
       {
           int a=d[minn], b=v[minn][j].first;
           if (d[v[minn][j].second]>a+b)
           {
               d[v[minn][j].second]=a+b;
               q.push({d[v[minn][j].second],v[minn][j].second});
           }
       }
    }
}
void afisare ()
{
   int i;
   for (i=2; i<=n; i++)
       if (d[i]==pinf)fout<<0<<' ';
       else fout<<d[i]<<' ';
}
int main()
{
    citire();
    dijkstra();
    afisare();
    fin.close();
    fout.close();
    return 0;

}