-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdijksta.cpp
More file actions
59 lines (56 loc) · 1.5 KB
/
dijksta.cpp
File metadata and controls
59 lines (56 loc) · 1.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include <iostream>
#define MAX 999
using namespace std;
void dijkstra(int** graph, int N, int E, int S);
int exmin(int* sss, int* key, int N);
int main(){
int N, E, S;
cin >> N >> E;
int** graph = new int* [E];
for (int i = 0; i < E; i ++){
graph[i] = new int [3];
}
for (int i = 0; i < E; i ++){
cin >> graph[i][0] >> graph[i][1] >> graph[i][2];
}
cin >> S;
dijkstra(graph, N, E, S);
}
void dijkstra(int** graph, int N, int E, int S){
int from;
int sss[N], key[N], parent[N];
for (int i = 0; i < N; i ++){
sss[i] = 0;
key[i] = MAX;
}
key[S-1] = 0;
parent[S-1] = 0;
for (int i = 0; i < N; i ++){
from = exmin(sss, key, N);
sss[from] = 1;
for (int i = 0; i < E; i ++){
if (graph[i][0]-1 == from && key[graph[i][1]-1] > key[graph[i][0]-1] + graph[i][2] && sss[graph[i][1]-1] == 0){
key[graph[i][1]-1] = key[graph[i][0]-1] + graph[i][2];
parent[graph[i][1]-1] = from+1;
}
}
}
for (int i = 0; i < N; i ++){
cout << key[i] << " ";
}
cout << endl;
for (int i = 0; i < N; i ++){
cout << parent[i] << " ";
}
}
int exmin(int* sss, int* key, int N){
int min_val = MAX;
int min_inx;
for (int i = 0; i < N; i ++){
if (key[i] < min_val && sss[i] == 0){
min_val = key[i];
min_inx = i;
}
}
return min_inx;
}