Pagini recente » Cod sursa (job #2180218) | Cod sursa (job #372851) | Cod sursa (job #1365302) | Cod sursa (job #1201035) | Cod sursa (job #2844738)
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
typedef pair < int, int > PII;
const int NMAX = 5e4 + 1;
const int INF = 1e9;
int N, M;
vector < PII > G[NMAX];
int d[NMAX];
bool Sel[NMAX];
auto cmp = [] (PII A, PII B)
{
if(!(A.first < B.first))
return 1;
return 0;
};
priority_queue < PII, vector < PII >, decltype(cmp) > H (cmp);
static inline void Read ()
{
f.tie(nullptr);
f >> N >> M;
for(int i = 1; i <= M; ++i)
{
int x = 0, y = 0, z = 0;
f >> x >> y >> z;
G[x].push_back({z, y});
G[y].push_back({z, x});
}
return;
}
static inline void Initialize (int S)
{
for(int i = 1; i <= N; ++i)
if(i != S)
d[i] = INF, Sel[i] = 0;
else
d[i] = 0, Sel[i] = 1;
return;
}
static inline void Dijkstra (int S)
{
Initialize(S);
for(auto it : G[S])
if(it.first < d[it.second])
d[it.second] = it.first, H.push({d[it.second], it.second});
while(!H.empty())
{
while(!H.empty() && Sel[H.top().second])
H.pop();
if(H.empty())
return;
int Node = H.top().second;
int dist = H.top().first;
Sel[Node] = 1, H.pop();
for(auto it : G[Node])
if(dist + it.first < d[it.second])
d[it.second] = dist + it.first, H.push({d[it.second], it.second});
}
return;
}
int main()
{
Read();
Dijkstra(1);
for(int i = 2; i <= N; ++i)
g << (d[i] == INF ? 0 : d[i]) << ' ';
g << '\n';
return 0;
}