JavaAlgorithms/Dijkshtra.java

57 lines
2.0 KiB
Java
Raw Normal View History

2017-04-24 21:57:05 +08:00
/*
@author : Mayank K Jha
*/
public class Solution {
2017-04-24 21:43:31 +08:00
public static void main(String[] args) throws IOException {
2017-04-24 21:55:18 +08:00
Scanner in =new Scanner(System.in);
2017-04-24 21:54:24 +08:00
int n=in.nextInt(); //n = Number of nodes or vertices
int m=in.nextInt(); //m = Number of Edges
long w[][]=new long [n+1][n+1]; //Adjacency Matrix
//Initializing Matrix with Certain Maximum Value for path b/w any two vertices
for (long[] row: w)
2017-04-24 21:43:31 +08:00
Arrays.fill(row, 1000000l);
2017-04-24 21:54:24 +08:00
//From above,we Have assumed that,initially path b/w any two Pair of vertices is Infinite such that Infinite = 1000000l
//For simplicity , We can also take path Value = Long.MAX_VALUE , but i have taken Max Value = 1000000l .
//Taking Input as Edge Location b/w a pair of vertices
2017-04-24 21:43:31 +08:00
for(int i=0;i<m;i++){
int x=in.nextInt(),y=in.nextInt();
long cmp=in.nextLong();
2017-04-24 21:54:24 +08:00
if(w[x][y]>cmp){ //Comparing previous edge value with current value - Cycle Case
w[x][y]=cmp; w[y][x]=cmp;
2017-04-24 21:43:31 +08:00
}
}
2017-04-24 21:54:24 +08:00
//Implementing Dijkshtra's Algorithm
2017-04-24 21:43:31 +08:00
Stack <Integer> t=new Stack<Integer>();
int src=in.nextInt();
for(int i=1;i<=n;i++){
if(i!=src){t.push(i);}}
Stack <Integer> p=new Stack<Integer>();
p.push(src);
w[src][src]=0;
while(!t.isEmpty()){int min=989997979,loc=-1;
for(int i=0;i<t.size();i++){
w[src][t.elementAt(i)]=Math.min(w[src][t.elementAt(i)],w[src][p.peek()]
+w[p.peek()][t.elementAt(i)]);
if(w[src][t.elementAt(i)]<=min){
min=(int) w[src][t.elementAt(i)];loc=i;}
}
p.push(t.elementAt(loc));t.removeElementAt(loc);}
2017-04-24 21:54:24 +08:00
//Printing shortest path from the given source src
2017-04-24 21:43:31 +08:00
for(int i=1;i<=n;i++){
if(i!=src && w[src][i]!=1000000l){System.out.print(w[src][i]+" ");}
2017-04-24 21:54:24 +08:00
else if(i!=src){System.out.print("-1"+" ");} //Printing -1 if there is no path b/w given pair of edges
2017-04-24 21:43:31 +08:00
}
2017-04-24 21:54:24 +08:00
2017-04-24 21:43:31 +08:00
}
2017-04-24 21:57:05 +08:00
}