算法训练 最短路 java_30-算法训练 最短路 spfa

对于10%的数据，n = 2，m = 2。

对于30%的数据，n <= 5，m <= 10。

对于100%的数据，1 <= n <= 20000，1 <= m <= 200000，-10000 <= l <= 10000，保证从任意顶点都能到达其他所有顶点。

提示： 下面的代码只有70分，超时了！

import java.util.ArrayList;

import java.util.LinkedList;

import java.util.Queue;

import java.util.Scanner;

import javax.management.Query;

public class Main {

public static Scanner cin = new Scanner(System.in);

public static ArrayList[] v1 = new ArrayList[20005];//v1[i]表示从i到达的点

public static ArrayList[] v2 = new ArrayList[20005];//v1对应路径的权值

public static int visit[] = new int[20005];

public static int d[] = new int[20005];

public static int n, m;

public static void main(String[] args) {

// TODO Auto-generated method stub

n = cin.nextInt();

m = cin.nextInt();

//初始化链表

for(int i = 1; i <= n; i++) {

v1[i] = new ArrayList();

v2[i] = new ArrayList();

}

int a, b, l;

for(int i = 0; i < m; i++) {

a = cin.nextInt();

b = cin.nextInt();

l = cin.nextInt();

v1[a].add(b); //有向图

v2[a].add(l);

}

spfa(1);

for(int i = 2; i <= n; i++) {

System.out.println(d[i]);

}

}

public static void spfa(int s) {

for(int i = 1; i <= n; i++) {

d[i] = 0x3f3f3f3f;

}

d[s] = 0;

visit[s] = 1;

int[] num = new int[20005];

Queue mq = new LinkedList<>();

mq.add(s);

num[s] = 1;

while(!mq.isEmpty()) {

int x = mq.peek();

mq.poll();

visit[x] = 0;

int ln = v1[x].size();

for(int i = 0; i < ln; i++) {

int v = (int) v1[x].get(i);

int len = (int) v2[x].get(i);

if(d[v] > d[x] + len) {

d[v] = d[x] + len;

if(visit[v] == 0) {

mq.add(v);

visit[v] = 1;

//num[v]++;

// if(num[v] > n)

// return;

}

}

}

}

}

}