forked from rajatgoyal715/Hackerrank
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDijkstra:ShortestReach2.java
74 lines (71 loc) · 1.94 KB
/
Dijkstra:ShortestReach2.java
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import java.io.*;
import java.util.*;
import java.text.*;
import java.math.*;
import java.util.regex.*;
public class Solution {
static int dist[];
static int n;
public static void main(String args[]){
Scanner sc=new Scanner(System.in);
int t=sc.nextInt();
while(t--!=0){
n=sc.nextInt();
int m=sc.nextInt();
dist=new int[n];
int a[][]=new int[n][n];
for(int i=0;i<m;i++){
int i1=sc.nextInt();
int i2=sc.nextInt();
int val=sc.nextInt();
if(a[i1-1][i2-1]!=0){
if(val<a[i1-1][i2-1]){
a[i1-1][i2-1]=val;
a[i2-1][i1-1]=val;
}
}
else{
a[i1-1][i2-1]=val;
a[i2-1][i1-1]=val;
}
}
int s=sc.nextInt()-1;
dijkstra(a,s);
for(int i=0;i<n;i++){
if(i!=s){
if(dist[i]==Integer.MAX_VALUE)
System.out.print("-1 ");
else
System.out.print(dist[i]+" ");
}
}
System.out.println();
}
}
public static void dijkstra(int[][] a,int s){
boolean set[] = new boolean[n];
for(int i=0;i<n;i++){
dist[i]=Integer.MAX_VALUE;
set[i]=false;
}
dist[s]=0;
for(int i=0;i<n-1;i++){
int u=minDistance(set);
set[u]=true;
for(int v=0;v<n;v++){
if(!set[v]&&a[u][v]!=0&&dist[u]!=Integer.MAX_VALUE&&dist[u]+a[u][v]<dist[v])
dist[v]=dist[u]+a[u][v];
}
}
}
public static int minDistance(boolean[] set){
int min=Integer.MAX_VALUE,min_index=0;
for(int i=0;i<n;i++){
if(set[i]==false&&dist[i]<=min){
min=dist[i];
min_index=i;
}
}
return min_index;
}
}