Cod sursa(job #2860967)

Utilizator Iordache_CezarIordache Cezar Iordache_Cezar Data 3 martie 2022 11:34:18
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.63 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <queue>
#define NMAX 50008
#define INF 1000000000ll
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
struct varf
{
    long long int x, c;
};

long long int n, x0=1, m;
vector<varf> G[NMAX];
bool uz[NMAX];
long long int cmin[NMAX];

class Compar
{
public:
  bool operator() (long long int a, long long int b) ///a si b sunt doua varfuri
  {
    ///min heap
    return cmin[a]>cmin[b];
  }
};
priority_queue<long long int, vector<long long int>, Compar>H;
void citire();
void dijkstra();
void afisare();
int main()
{
    citire();
    dijkstra();
    afisare();
    return 0;
}
void citire()
{
    long long int i, j, c;
    varf aux;
    fin>>n>>m;
    for (int k = 1; k <= m; k++)
    {
        fin >> i >> aux.x >> aux.c;
        G[i].push_back(aux);
    }
    for(i=0; i<=n; i++)
        cmin[i]=INF;
    cmin[x0]=0;
}
void dijkstra()
{
    long long int i, j, minim, vfmin;
    H.push(x0);
    while (!H.empty())
    {
        vfmin=H.top();
        H.pop();
        minim=cmin[vfmin];
        ///optimizez eventual costurile minime
        for(j=0; j<G[vfmin].size(); j++)
        {
            long long vf = G[vfmin][j].x;
            long long cost = G[vfmin][j].c;
            if(cmin[vf]>minim+cost)
            {
                cmin[vf]=minim+cost;
                H.push(vf);
            }
        }
    }
}
void afisare()
{
    long long int i, j, cnt;
    for(i=2; i<=n; i++)
        if(cmin[i]==INF)
            fout<<0<<' ';
           else
            fout<<cmin[i]<<' ';
}