Java实现 蓝桥杯 算法提高 道路和航路

问题描述
农夫约翰正在针对一个新区域的牛奶配送合同进行研究。他打算分发牛奶到T个城镇(标号为1…T),这些城镇通过R条标号为(1…R)的道路和P条标号为(1…P)的航路相连。

每一条公路i或者航路i表示成连接城镇Ai(1<=A_i<=T)和Bi(1<=Bi<=T)代价为Ci。每一条公路,Ci的范围为0<=Ci<=10,000;由于奇怪的运营策略,每一条航路的Ci可能为负的,也就是-10,000<=Ci<=10,000。

每一条公路都是双向的,正向和反向的花费是一样的,都是非负的。

每一条航路都根据输入的Ai和Bi进行从Ai->Bi的单向通行。实际上,如果现在有一条航路是从Ai到Bi的话,那么意味着肯定没有通行方案从Bi回到Ai。

农夫约翰想把他那优良的牛奶从配送中心送到各个城镇,当然希望代价越小越好,你可以帮助他嘛?配送中心位于城镇S中(1<=S<=T)。

输入格式
输入的第一行包含四个用空格隔开的整数T,R,P,S。

接下来R行,描述公路信息,每行包含三个整数,分别表示Ai,Bi和Ci。

接下来P行,描述航路信息,每行包含三个整数,分别表示Ai,Bi和Ci。

输出格式
输出T行,分别表示从城镇S到每个城市的最小花费,如果到不了的话输出NO PATH。
样例输入
6 3 3 4
1 2 5
3 4 5
5 6 10
3 5 -100
4 6 -100
1 3 -10
样例输出
NO PATH
NO PATH
5
0
-95
-100
数据规模与约定
对于20%的数据,T<=100,R<=500,P<=500;

对于30%的数据,R<=1000,R<=10000,P<=3000;

对于100%的数据,1<=T<=25000,1<=R<=50000,1<=P<=50000。

2 解决方案
本题主要考查最短路径,其中时间效率最好的算法为SPFA算法,但是下面的代码在蓝桥系统中评分为30或者35分,原因:运行超时,而同样的方法,用C来实现在蓝桥系统中评分为95或者100分,用C实现的代码请见文末参考资料1。

如果有哪位同学Java版本代码测评分数超过50分的同学,还望借鉴一下代码,不甚感激~

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static int T, R, P, S;
    public static ArrayList<edge>[] map;
    public static int[] distance;
    
    
    static class edge {
        public int a;  //边的起点
        public int b;  //边的终点
        public int v;  //边的权值
        
        public edge(int a, int b, int v) {
            this.a = a;
            this.b = b;
            this.v = v;
        }
    }
    
    @SuppressWarnings("unchecked")
    public void init() {
        map = new ArrayList[T + 1];
        distance = new int[T + 1];
        for(int i = 0;i <= T;i++) {
            map[i] = new ArrayList<edge>();
            distance[i] = Integer.MAX_VALUE;
        }
    }
    
    public void spfa() {
        ArrayList<Integer> town = new ArrayList<Integer>();
        distance[S] = 0;
        town.add(S);
        int[] count = new int[T + 1];
        boolean[] visited = new boolean[T + 1];
        count[S]++;
        visited[S] = true;
        while(town.size() != 0) {
            int start = town.get(0);
            town.remove(0);
            visited[start] = false;
            for(int i = 0;i < map[start].size();i++) {
                edge to = map[start].get(i);
                if(distance[to.b] > distance[start] + to.v) {
                    distance[to.b] = distance[start] + to.v;
                    if(!visited[to.b]) {
                        town.add(to.b);
                        visited[to.b] = true;
                        count[to.b]++;
                        if(count[to.b] > T)  //此时有负环出现
                            return;
                    }
                }
            }
        }
    }
    
    public void getResult() {
        spfa();
        for(int i = 1;i <= T;i++) {
            if(distance[i] == Integer.MAX_VALUE)
                System.out.println("NO PATH");
            else
                System.out.println(distance[i]);
        }
    }
    
    
    public static void main(String[] args) {
        Main test = new Main();
        Scanner in = new Scanner(System.in);
        T = in.nextInt();
        R = in.nextInt();
        P = in.nextInt();
        S = in.nextInt();
        test.init();
        for(int i = 1;i <= R;i++) {
            int a = in.nextInt();
            int b = in.nextInt();
            int v = in.nextInt();
            map[a].add(new edge(a, b, v));
            map[b].add(new edge(b, a, v));
        }
        for(int i = 1;i <= P;i++) {
            int a = in.nextInt();
            int b = in.nextInt();
            int v = in.nextInt();
            map[a].add(new edge(a, b, v));
        }
        test.getResult();
    }
}
posted @ 2019-07-26 18:26  南墙1  阅读(27)  评论(0编辑  收藏  举报