Pagini recente » Cod sursa (job #627829) | Cod sursa (job #2747485) | Cod sursa (job #1318467) | Cod sursa (job #517517) | Cod sursa (job #2614005)
#include <iostream>
#include <fstream>
#include <vector>
#include <set>
#include <climits>
#define N_MAX 50001
using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
struct edge
{
int node, cost;
};
struct nodeCost
{
int node, cost;
};
struct classcomp
{
bool operator()(nodeCost a, nodeCost b) const
{
return a.cost < b.cost || a.cost == b.cost && a.node < b.node;
}
};
int n, m, a, b, v;
vector<edge> graph[N_MAX];
vector<int> calculateCosts()
{
set<nodeCost, classcomp> costs;
vector<int> finalCosts(n, INT_MAX);
costs.insert({0, 0});
for (int i = 1; i < n; i++)
{
costs.insert({i, INT_MAX});
}
while (costs.size() > 0)
{
auto first = *(costs.begin());
costs.erase(costs.begin());
finalCosts[first.node] = first.cost;
for (auto d : graph[first.node])
{
int lungNou = first.cost + d.cost;
if (finalCosts[d.node] > lungNou || finalCosts[d.node] == INT_MAX)
{
costs.erase({d.node, finalCosts[d.node]});
costs.insert({d.node, lungNou});
finalCosts[d.node] = lungNou;
}
}
}
return finalCosts;
}
int main()
{
in >> n >> m;
for (int i = 0; i < m; i++)
{
in >> a >> b >> v;
graph[a - 1].push_back({b - 1, v});
}
auto aa = calculateCosts();
for (int i = 1; i < aa.size(); i++)
{
if (aa[i] != INT_MAX)
{
out << aa[i] << ' ';
}
else
{
out << "0 ";
}
}
return 0;
}