Cod sursa(job #3296694)

Utilizator Benjamin4321234Benjamin Secara Benjamin4321234 Data 15 mai 2025 16:32:14
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
#define NMAX 50001
int n,m,x,y,z;
int dist_min[NMAX];
bool vizitat[NMAX];

class cmp
{
public:
    bool operator ()(int a, int b)
    {
        return dist_min[a]>dist_min[b];
    }
};

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

void dijkstra(int m)
{

    for(int i=1; i<=n; i++)
    {
        dist_min[i]=100000001;
    }

    q.push(m);
    vizitat[m]=1;
    dist_min[m]=0;

    while(!q.empty())
    {
        int x=q.top();
        q.pop();
        vizitat[x]=0;
        for(auto u:v[x])
        {
            if(dist_min[x]+u.second<dist_min[u.first])
            {
                dist_min[u.first]=dist_min[x]+u.second;
                if(!vizitat[u.first])
                {
                    vizitat[u.first]=1;
                    q.push(u.first);
                }
            }
        }
    }
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=m; i++)
    {
        fin>>x>>y>>z;
        v[x].push_back({y,z});
    }
    dijkstra(1);
    for(int i=2; i<=n; i++)
    {
        if(dist_min[i]!=100000001)
        {
            fout<<dist_min[i]<<" ";
        }
        else
        {
            fout<<0<<" ";
        }
    }
    return 0;
}