传送门:
P4779 【模板】单源最短路径(标准版) - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)https://www.luogu.com.cn/problem/P4779
先给代码:
import java.util.*;
import java.io.*;
class edge implements Comparable<edge>{
int pos;
int dis;
public edge(int pos,int dis){
this.pos=pos;
this.dis=dis;
}
@Override
public int compareTo(edge o){
return this.dis-o.dis;
}
}
public class Main {
static int n,m,s,cnt;
static int[] vv,ww,nex,head;
static edge[] e;
public static void main(String[] args) throws IOException{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StreamTokenizer st = new StreamTokenizer(br);
st.nextToken();n=(int)st.nval;
st.nextToken();m=(int)st.nval;
st.nextToken();s=(int)st.nval;
vv=new int[m+1];ww=new int[m+1];head=new int[n+1];nex=new int[m+1];
e=new edge[m+1];
for(int i=1;i<=m;i++){
st.nextToken();int u=(int)st.nval;
st.nextToken();int v=(int)st.nval;
st.nextToken();int w=(int)st.nval;
add_edge(u,v,w);
}
int[] ans = dijkstra(s);
for(int i=1;i<=n;i++){
System.out.print(ans[i] + " ");
}
}
static void add_edge(int u,int v,int w){
vv[++cnt]=v;
ww[cnt]=w;
nex[cnt]=head[u];
head[u]=cnt;
}
static int[] dijkstra(int start){
boolean[] vis=new boolean[n+1];
Arrays.fill(vis,false);
Queue<edge> q = new PriorityQueue<>();
q.offer(new edge(start,0));
int[] dis=new int[n+1];
Arrays.fill(dis,0x7fffffff);
dis[start]=0;
while(!q.isEmpty()){
edge node = q.poll();
int x=node.pos;
if(vis[x]) continue;
vis[x]=true;
for(int i=head[x];i>0;i=nex[i]){
int y=vv[i];
if(dis[y]>dis[x]+ww[i]){
dis[y]=dis[x]+ww[i];
if(!vis[y])
q.offer(new edge(y,dis[y]));
}
}
}
return dis;
}
}
这道题需要用到堆优化,也就是优先队列,c++版的过程可以看这位大佬的dijkstra 详解 - little_sun 的博客 - 洛谷博客 (luogu.com.cn)https://www.luogu.com.cn/blog/little-sun/dijkstra
建议懂了他的做法之后,再看我的代码,会好理解一些。但是特别需要注意的是,c++可以用struct,但java不可以用class,为什么呢?因为java的类太笨了,做题要么超时,要么超内存。
这时候我们可以为每一个变量定义一个数组,也是一样的道理
vv=new int[m+1];ww=new int[m+1];head=new int[n+1];nex=new int[m+1];