洛谷 P1119 灾后重建（java实现）

P1119 灾后重建

题目背景

B地区在地震过后，所有村庄都造成了一定的损毁，而这场地震却没对公路造成什么影响。但是在村庄重建好之前，所有与未重建完成的村庄的公路均无法通车。换句话说，只有连接着两个重建完成的村庄的公路才能通车，只能到达重建完成的村庄。

题目描述

给出B地区的村庄数N，村庄编号从0到N-1，和所有M条公路的长度，公路是双向的。并给出第i个村庄重建完成的时间t_i，你可以认为是同时开始重建并在第t_i天重建完成，并且在当天即可通车。若t_i为0则说明地震未对此地区造成损坏，一开始就可以通车。之后有Q个询问(x, y, t)，对于每个询问你要回答在第t天，从村庄x到村庄y的最短路径长度为多少。如果无法找到从x村庄到y村庄的路径，经过若干个已重建完成的村庄，或者村庄x或村庄y在第t天仍未重建完成 ，则需要返回-1。

输入格式

第一行包含两个正整数N,M，表示了村庄的数目与公路的数量。

第二行包含N个非负整数t_0, t_1,…, t_{N-1}，表示了每个村庄重建完成的时间，数据保证了t_0 ≤ t_1 ≤ … ≤ t_{N-1}。

接下来M行，每行3个非负整数i, j, w，w为不超过10000的正整数，表示了有一条连接村庄i与村庄j的道路，长度为w，保证i≠j，且对于任意一对村庄只会存在一条道路。

接下来一行也就是M+3行包含一个正整数Q，表示Q个询问。

接下来Q行，每行3个非负整数x, y, t，询问在第t天，从村庄x到村庄y的最短路径长度为多少，数据保证了t是不下降的。

输出格式

共Q行，对每一个询问(x, y, t)输出对应的答案，即在第t天，从村庄x到村庄y的最短路径长度为多少。如果在第t天无法找到从x村庄到y村庄的路径，经过若干个已重建完成的村庄，或者村庄x或村庄y在第t天仍未修复完成，则输出-1。

输入输出样例

输入 #1

4 5
1 2 3 4
0 2 1
2 3 1
3 1 2
2 1 4
0 3 5
4
2 0 2
0 1 2
0 1 3
0 1 4

输出 #1

-1
-1
5
4

说明/提示

对于30%的数据，有N≤50；

对于30%的数据，有t_i= 0，其中有20%的数据有t_i = 0且N>50；

对于50%的数据，有Q≤100；

对于100%的数据，有N≤200，M≤N×(N−1)/2，Q≤50000，所有输入数据涉及整数均不超过100000。

题解一

package p1119;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
import java.util.Arrays;
import java.util.PriorityQueue;
//	dijkstra 60
public class Main2{
	static int N = 210,INF=0x3f3f3f3f;
	static int n,m;
	static int g[][] = new int[N][N];
	static int rt[] = new int[N];
	static int dist[] = new int[N];
	static boolean st[] = new boolean[N];
	public static void dijkstra(int q,int time) {
		PriorityQueue<Pair> p = new PriorityQueue<>();
		Arrays.fill(dist,INF);
		Arrays.fill(st, false);
		p.add(new Pair(q,0));
		dist[q] = 0;
		while(!p.isEmpty()) {
			Pair pair = p.poll();
			int t = pair.x,dis = pair.y;
			if(st[t]) continue;
			st[t] = true;
			for(int i=0;i<n;i++) {
				if(g[t][i]==INF||rt[i]>time) continue;
				if(dist[i] > dis + g[t][i]) {
					dist[i] = dis + g[t][i];
					p.add(new Pair(i,dist[i]));
				}
			}
		}
	}
	public static void main(String[] args) throws Exception{
		Read r = new Read();
		n = r.nextInt();m = r.nextInt();
		for(int i =0;i<n;i++) {
			Arrays.fill(g[i], INF);
		}
		for(int i=0;i<n;i++) {
			rt[i] = r.nextInt();
		}
		for(int i=0;i<m;i++) {
			int a = r.nextInt(),b = r.nextInt(),c = r.nextInt();
			g[a][b]=c;g[b][a]=c;
		}
		int Q = r.nextInt();
		while(Q-->0) {
			int a = r.nextInt(),b = r.nextInt(),c = r.nextInt();
			if(rt[a]>c||rt[b]>c) {
				System.out.println(-1);
				continue;
			}
			dijkstra(a,c);
			if(dist[b]==INF) {
				System.out.println(-1);
			}else {
				System.out.println(dist[b]);
			}
		}
	}
}
class Pair implements Comparable<Pair>{
	int x,y;
	public Pair(int a,int b){
		x=a;y=b;
	}
	@Override
	public int compareTo(Pair o) {
		return this.y - o.y;
	}
}
class Read {
	StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
	public int nextInt() throws Exception{
		st.nextToken();
		return (int)st.nval;
	}
}

题解二

package p1119;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
public class Main3{
	static int N = 210,INF=0x3f3f3f3f;
	static int n,m;
	static int d[][] = new int[N][N];
	static int rt[] = new int[N];
	public static void floyd(int k) {
		for(int i=0;i<n;i++) {
			for(int j=0;j<n;j++) {
				d[i][j] = Math.min(d[i][j],d[i][k]+d[k][j]);
			}
		}
	}
	public static void main(String[] args) throws Exception{
		Read r = new Read();
		n = r.nextInt();m = r.nextInt();
		for(int i=0;i<n;i++) {
			for(int j=0;j<n;j++) {
				if(i==j) {
					d[i][j] = 0;
				}else {
					d[i][j] = INF;
				}
			}
		}
		for(int i=0;i<n;i++) {
			rt[i] = r.nextInt();
		}
		for(int i=0;i<m;i++) {
			int a = r.nextInt(),b = r.nextInt(),c = r.nextInt();
			d[a][b]=c;d[b][a]=c;
		}
		int Q = r.nextInt();
		int now = 0;
		while(Q-->0) {
			int a = r.nextInt(),b = r.nextInt(),c = r.nextInt();
			while(rt[now]<=c&&now<n) {	//	依次更新点，使它可以被用来更新其他的点 
				floyd(now);
				now++;
			}
			if(rt[a]>c||rt[b]>c) {
				System.out.println(-1);
			}else {
				if(d[a][b]==INF) {
					System.out.println(-1);
				}else {
					System.out.println(d[a][b]);
				}
			}
		}
	}
}
class Read {
	StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
	public int nextInt() throws Exception{
		st.nextToken();
		return (int)st.nval;
	}
}