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