洛谷 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,MN×(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;
	}
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

杜柠函

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值