-
Notifications
You must be signed in to change notification settings - Fork 0
/
lanqiao 算法训练最短路 SPFA实现 超时 40分
71 lines (57 loc) · 1.67 KB
/
lanqiao 算法训练最短路 SPFA实现 超时 40分
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
public class Main{
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n=in.nextInt();
int m=in.nextInt();
int graph[][] = new int[n][n];
int dis[] = new int[n];
int count[] = new int[n];
boolean used[] = new boolean[n];
for(int i=0;i<n;i++) {//原来的话graph数组没有初始化,所以对后来spfa算法里if的判断条件造成了影响
for(int j=0;j<n;j++) {
graph[i][j] = Integer.MAX_VALUE;
}
}
for(int i=0;i<n;i++) {
int r=in.nextInt();
int c=in.nextInt();
graph[r-1][c-1] = in.nextInt();//数组从0开始,数据从1开始
}
for(int i=0;i<n;i++) {
dis[i] = i==0?0:Integer.MAX_VALUE;规定起点是0
}
spfa(dis,graph,used,count);
}
public static void spfa(int dis[],int graph[][],boolean used[],int count[]) {
Queue<Integer> q= new LinkedList<Integer>();
q.add(0);
used[0] = true;
count[0]++;
int tempt=0;
while(!q.isEmpty()) {
int head = q.peek();//原来q.poll()及Used[head]=false是在这里的,给挪到后面去了
for(int i=0;i<dis.length;i++) {
if(i!=head&&graph[head][i]!=Integer.MAX_VALUE) {
dis[i] = dis[i]>dis[head]+graph[head][i]?dis[head]+graph[head][i]:dis[i];
if(!used[i]) {//原来上面的dis[i]=……是在if条件里面,逻辑好像有些问题
used[i] = true;
q.add(i);
count[i]++;
tempt=i;
}
}
}
if(count[tempt]>dis.length) {//添加的判断条件
break;
}
q.poll();
used[head] = false;
}
for(int k=1;k<dis.length;k++) {
System.out.println(dis[k]);
}
}
}