Pagini recente » Istoria paginii runda/simulare_fmi_nostress_nerezolvate/clasament | Cod sursa (job #3271384)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
void dijkstra(int startNode, vector<vector<pair<int, int>>> &costMatrix, vector<bool> &visited, vector<int> &answer)
{
queue<int> nodes;
nodes.push(startNode);
while (!nodes.empty())
{
int node = nodes.front();
nodes.pop();
for (auto it = costMatrix[node].begin(); it != costMatrix[node].end(); it++)
if (!visited[it->first])
{
answer[it->first] = min(answer[it->first], answer[node] + it->second);
nodes.push(it->first);
}
visited[node] = true;
}
}
int main()
{
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
int nodeCount, vertexCount, source, destination, cost;
in >> nodeCount >> vertexCount;
vector<vector<pair<int, int>>> costMatrix(nodeCount + 1, vector<pair<int, int>>());
vector<bool> visited(nodeCount + 1, false);
vector<int> answer(nodeCount + 1, 1e9);
for (int vertexIndex = 1; vertexIndex <= vertexCount; vertexIndex++)
{
in >> source >> destination >> cost;
costMatrix[source].push_back(pair<int, int>(destination, cost));
}
answer[1] = 0;
dijkstra(1, costMatrix, visited, answer);
for (int nodeIndex = 2; nodeIndex < answer.size(); nodeIndex++)
out << answer[nodeIndex] << ' ';
return 0;
}