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;
}
}