import java.util.Arrays;
import java.util.Scanner;
//迪杰斯特拉算法模版
public class DijkstraMain {
static Scanner in = new Scanner(System.in);
static int[] dis = new int[505];//表示起点到某点的最短距离
static boolean[] vis = new boolean[505];//判断某点是否被访问
static int[][] e = new int[505][505];//存放图的数组
static int[] pre = new int[505];//存放某点的前驱节点
static int n;//n个顶点
static int m;//m条边
static int INF = 999999;
static void dijstra(int v)
{
for(int i=0;i<n;i++)
{
dis[i] = e[v][i];
pre[i] = v;
}
vis[v] = true;
//循环结束条件为n-1的原因是已经访问过起点v了,还剩下n-1个顶点没有访问
for(int i=0;i<n-1;i++)
{
int min = INF;
int index = -1;
for(int j=0;j<n;j++)
{
if(!vis[j]&&min>dis[j])
{
min = dis[j];
index = j;
}
}
vis[index] = true;
//更新dis数组
for(int j=0;j<n;j++)
{
if(!vis[j]&&e[index][j]+dis[index]<dis[j])
{
dis[j] = e[index][j]+dis[index];
pre[j] = index;
}
}
}
}
public static void main(String[] args) {
int v,u;
n = in.nextInt();//输入的顶点数
m = in.nextInt();//输入的边数
for(int i=0;i<n;i++)
Arrays.fill(e[i],INF);
for(int i=0;i<m;i++)//初始化图
{
int a = in.nextInt();//输入边的起点
int b = in.nextInt();//输入边的终点
int edgelength = in.nextInt();//输入边的长度
e[a][b] = e[b][a] = edgelength;
}
v = in.nextInt();//从哪个起点开始
u = in.nextInt();//起点v到某点u的最短距离
dijstra(v);
System.out.println("最短距离为:" + dis[u]);
}
}
//测试样例
/*
6 8
0 1 1
0 3 4
0 4 4
1 3 2
2 5 1
3 2 2
3 4 3
4 5 3
0
1
*/
Dijkstra算法模版
最新推荐文章于 2023-08-06 15:45:33 发布