Cod sursa(job #2775169)

Utilizator RK13Barbu Eduard RK13 Data 14 septembrie 2021 18:27:55
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.05 kb
#include <bits/stdc++.h>
#define INF 2000000000
using namespace std;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

struct elem
{int nod,c;
 bool operator < (const elem &a) const
    {
        return c>a.c;
    }
};

vector<elem>a[50002];
priority_queue<elem>pq;
int sol[50002];
int main()
{
int n,m,x,y,z,i,j,l;
elem p;
f>>n>>m;
for (i=1;i<=m;i++)
{f>>x>>y>>z;
a[x].push_back({y,z});
a[y].push_back({y,z});

}

for (i=2;i<=n;i++) sol[i]=INF;
pq.push({1,0});

while (!pq.empty())
{p=pq.top();
pq.pop();
if (p.c==sol[p.nod]) {l=a[p.nod].size();
                    for (i=0;i<l;i++) if (sol[a[p.nod][i].nod]>sol[p.nod]+a[p.nod][i].c) {sol[a[p.nod][i].nod]=sol[p.nod]+a[p.nod][i].c;
                                                                                    pq.push({a[p.nod][i].nod,sol[a[p.nod][i].nod]});
                                                                                   }
                   }
}
for (i=2;i<=n;i++) if (sol[i]==INF) g<<0<<' ';
                        else g<<sol[i]<<' ';
}