Pagini recente » Cod sursa (job #2261946) | Cod sursa (job #3150205) | Cod sursa (job #1832044) | Cod sursa (job #123914) | Cod sursa (job #2384953)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <climits>
int main()
{
std::ifstream fin;
fin.open("dijkstra.in");
std::ofstream fout;
fout.open("dijkstra.out");
int verticesCount, arcsCount;
fin>>verticesCount>>arcsCount;
std::vector< std::pair<int,int> > adjList[50001];
std::priority_queue< std::pair<int,int> > q;
int arcFrom, arcTo, cost;
for(int i=1;i<=arcsCount;i++)
{
fin>>arcFrom>>arcTo>>cost;
adjList[arcFrom].push_back(std::make_pair(cost,arcTo) );
}
int vis[50001] = {0};
int costList[50001];
for (int i=1; i <= verticesCount; i++)
{
costList[i] = INT_MAX;
}
q.push(std::make_pair(0,1));
costList[1] = 0;
while (!q.empty()) {
std::pair<int,int> curentVertex;
curentVertex = q.top();
curentVertex.first = curentVertex.first * -1;
q.pop();
if(!vis[curentVertex.second])
{
vis[curentVertex.second] = 1;
for (auto const& vertex: adjList[curentVertex.second])
{
int curentCost = curentVertex.first + vertex.first;
//int curentCost = costList[curentVertex.second] + vertex.first;
if( costList[vertex.second] > curentCost)
{
costList[vertex.second] = curentCost;
q.push(std::make_pair(vertex.first * -1,vertex.second));
}
}
}
}
for (int i=2;i<=verticesCount;i++)
if(costList[i] != INT_MAX) fout<< costList[i] << " ";
else fout<<"0 ";
return 0;
}