Dijkstra求最短路(PriorityQueue)

给定一个n个点m条边的有向图,图中可能存在重边和自环,所有边权均为非负值。

请你求出1号点到n号点的最短距离,如果无法从1号点走到n号点,则输出-1。

输入格式
第一行包含整数n和m。

接下来m行每行包含三个整数x,y,z,表示存在一条从点x到点y的有向边,边长为z。

输出格式
输出一个整数,表示1号点到n号点的最短距离。

如果路径不存在,则输出-1。

数据范围
1≤n,m≤1.5×105,
图中涉及边长均不小于0,且不超过10000。

输入样例:
3 3
1 2 2
2 3 1
1 3 4
输出样例:
3

//package No1;

import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Scanner;

public class Main {
     

	static int N = 150010;
	static PriorityQueue<tt> q = new PriorityQueue(N, new Comparator<tt>() {
     
		public int compare(tt a, tt b) {
     
			return a.dis - b.dis;
		}
	});
	static int e[] = new int[N];
	static int ne[] = new int[N];
	static int h[] = new int[N];
	static int idx, n, m;
	static int w[] = new int[N];
	static int dis[] = new int[N];
	static int vis[] = new int[N];

	public static void main(String[] args) {
     
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		for (int i = 1; i <= n; i++) {
     
			h[i] = -1;
			dis[i] = Integer.MAX_VALUE;
		}

		m = in.nextInt();
		for (int i = 0; i < m; i++) {
     
			int a=in.nextInt();
			int b=in.nextInt();
			int val=in.nextInt();
			add(a, b, val);
		}
		System.out.println(dijskra());

	}

	static int dijskra() {
     
		dis[1] = 0;
		q.add(new tt(1, 0));
		for (int i = 1; i <= n; i++) {
     

			tt t = q.poll();
			if (t==null)
				break;
			vis[t.index] = 1;

			for (int j = h[t.index]; j != -1; j = ne[j]) {
     
				int index = e[j];
				int d = w[j];
				if (vis[index] == 0) {
     
					dis[index] = Math.min(dis[index], d + dis[t.index]);

					q.add(new tt(index, dis[index]));
				}
			}
		}
		if (dis[n] == Integer.MAX_VALUE)
			return -1;
		return dis[n];
	}

	static void add(int a, int b, int val) {
     
		e[idx] = b;
		w[idx] = val;
		ne[idx] = h[a];
		h[a] = idx++;
	}
}

class tt {
     
	int index;
	int dis;

	public tt(int index, int dis) {
     
		this.index = index;
		this.dis = dis;
	}
}

你可能感兴趣的:(ACM)