L2-001 紧急救援 (25分)java

作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。

输入格式:

输入第一行给出4个正整数N、M、S、D,其中N(2≤N≤500)是城市的个数,顺便假设城市的编号为0 ~ (N−1);M是快速道路的条数;S是出发地的城市编号;D是目的地的城市编号。

第二行给出N个正整数,其中第i个数是第i个城市的救援队的数目,数字间以空格分隔。随后的M行中,每行给出一条快速道路的信息,分别是:城市1、城市2、快速道路的长度,中间用空格分开,数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式:

第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔,输出结尾不能有多余空格。

输入样例:

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

输出样例:

2 60
0 1 3

代码:java最后一点超时

import java.util.Scanner;

public class Main{
    private static int[] dis,w,pre,num,weight;
    private static boolean[] vis;
    private static int n,m,s,d;
    private static int[][] G;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();m = sc.nextInt();
        s = sc.nextInt();d = sc.nextInt();
        dis = new int[n];vis = new boolean[n];w=new int[n];pre=new int[n];num=new int[n];
        weight = new int[n];
        G = new int[n][n];//邻接矩阵
        for(int i = 0; i<n; i++){
            weight[i] = sc.nextInt();
            dis[i] = 1000;//设起点到达各点的最短距离
        }
        for(int i = 0; i<n; i++){
            for(int j = 0; j<n; j++){
                if(i != j)
                    G[i][j] = 1000;
            }
        }
        for(int i = 0; i<m; i++){
            int a = sc.nextInt();int b = sc.nextInt();int c = sc.nextInt();
            G[a][b] = c;
            G[b][a] = c;
        }
        Dijkstra(s);
        System.out.println( num[d]+" "+w[d]);
        dfs(s,d);
        System.out.println();

    }
    private static void dfs(int s,int v){
        if(s == v){
            System.out.print(s);
            return;
        }
        dfs(s,pre[v]);
        System.out.print(" "+v);
    }

    private static void Dijkstra(int s) {
        dis[s] = 0;//将起点到达自身的距离修改为0;
        num[s] = 1;//最短路径条数初始化为1;
        w[s] = weight[s];//救援队数目修改为起点城市的救援队数目
        for (int i = 0; i<n; i++){
            int u = -1,min = 1000;
            for(int j = 0; j<n; j++){
                if(!vis[j] && dis[j] < min){
                    min = dis[j];
                    u = j;
                }
            }
            if(u == -1)return;
            vis[u] = true;
            for(int v = 0; v<n; v++){
                if(!vis[v] && G[u][v] != 1000 && dis[u] + G[u][v] < dis[v]){
                    dis[v] = dis[u] + G[u][v];
                    w[v] = w[u]+weight[v];
                    num[v] = num[u];
                    pre[v] = u;
                }else if(!vis[v] && G[u][v] != 1000 &&dis[u] + G[u][v] == dis[v]){
                    num[v] += num[u];//最短距离相同时路径条数累加
                    if(w[v] < w[u]+weight[v]){
                        w[v] = w[u]+weight[v];
                        pre[v] = u;
                    }
                }
            }
        }

    }
}

猜你喜欢

转载自blog.csdn.net/qq_44985985/article/details/107881424