Cod sursa(job #3201853)

Utilizator stefan2806Radulescu Stefan stefan2806 Data 9 februarie 2024 21:18:47
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <fstream>
#include <vector>
#include <queue>
#include <limits>

using namespace std;

#define pii pair<int,int>

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

int n,m,i,x,y,c,d[50010],viz[50010];

struct cmp
{
    bool operator()(pii x,pii y)
    {
        return x.second>y.second;
    }
};

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

void dijkstra(int x)
{
    int inf=2000000000;
    int vecin,cost;
    for(i=1; i<=n; i++)
        d[i]=inf;
    d[x]=0;
    q.push({x,0});
    while(!q.empty())
    {
        x=q.top().first;
        q.pop();
        if(viz[x]==1)
            continue;
        viz[x]=1;
        for(auto it:v[x])
        {
            vecin=it.first;
            cost=it.second;
            if(d[vecin]>d[x]+cost)
            {
                d[vecin]=d[x]+cost;
                q.push({vecin,d[x]+cost});
            }
        }
    }

    for(i=2; i<=n; i++)
        if(d[i]==inf)cout<<"0 ";
        else
            cout << d[i]<<' ';
}

int main()
{
    cin>>n>>m;
    for(i=1; i<=m; i++)
    {
        cin>>x>>y>>c;
        v[x].push_back({y,c});
    }
    dijkstra(1);
    return 0;
}