java优先队列实现dijskra

参考代码:

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Scanner cin = new Scanner(System.in);
		int n = cin.nextInt();
		int k = cin.nextInt();
		int[] dist = new int[n];
		boolean[] used = new boolean[n];
		List<List<Integer>> edge = new ArrayList<>();
		List<List<Integer>> list = new ArrayList<>();
		PriorityQueue<int[]> q = new PriorityQueue<>(new Comparator<int[]> (){
		    public int compare(int[] o1, int[] o2) {
		        return o1[0] - o2[0];
		    }
		});
		for(int i = 0; i < n; i++) {
		    dist[i] = (int) 1e9;
		    list.add(new ArrayList<Integer>());
		    edge.add(new ArrayList<Integer>());
		}
		for(int i = 0; i < n; i++) {
		    int x = i;
		    List<Integer> l = list.get(x);
		    List<Integer> e = edge.get(x);
		    for(int j = 0; j < n; j++) {
		        int z = cin.nextInt();
		        if(i == j) continue;
		        if(z > 0){
		            l.add(j);
		            e.add(z);
		        }
		    }
		    list.set(x, l);
		    edge.set(x, e);
		}
		dist[k] = 0;
		q.add(new int[] {0, k});
		while(!q.isEmpty()) {
		    int[] now = q.poll();
		    int x = now[1];
		    if(used[x]) continue;
		    used[x] = true;
		    for(int i = 0; i < list.get(x).size(); i++) {
		        int y = list.get(x).get(i);
		        int z = edge.get(x).get(i);
		        if(dist[x] + z < dist[y]) {
		            dist[y] = dist[x] + z;
		            q.add(new int[]{dist[y], y});
		        }
		    }
		}
		for(int i = 0; i < n; i++) {
		    if(k != i) {
		        int ans = (dist[i] == (int) 1e9 ? -1 : dist[i]);
		        System.out.print(ans + " ");
		    }
		    
		}
		
		
	}
}