贪心算法-单源最短路径

算法思想:贪心算法

实际问题:单源最短路径

编程语言:Java

问题描述

  单源最短路径算法,又称 迪杰斯特拉算法 。其目的是寻找从一个顶点到其余各顶点的最短路径算法,解决的是有权图中最短路径问题。

算法构造

相关解释

  • 观测域:假设起点为v点,观测域便为v点的四周,即v的所有邻接点;
  • 点集 V:图中所有点的集合;
  • 点集 S:已经找到最短路径的终点集合;
  • 数组 D:存储观测域内能观测到的最短路径,算上起点一共 n 个数值。比如 D[k] 对应在观测域中能观测到的,到顶点 k 的最短路径;
  • 邻接矩阵 a:存储着有权图中的边的信息,是一个二维数组。比如 a[1][2] = 5 表示在有权图中,点 1 和点 2 之间有边,且边的权值为 5。如果两点之间没边,则用负数或则无穷大(∞)表示。

算法步骤

  • 第一步:初始化点集 S,将起点 v 收入 S 中。初始化数组 D:D[k] = a[v][k];
  • 第二步:找寻次短路径。即查找数组 D,找出观测域中最短路径(v, j):D[j] = min(D[k] | k 不属于 S)。将点 j 加入点集 S 中;
  • 第三步:将 j 的邻接点并入观测域,即用 j 的邻接点更新数组 D;
  • 第四步:不断重复第二步和第三步,直到节点全部压入 S 中为止。

注:贪心算法的思想主要就体现在第二步和第三步之中。

Java 代码

  本代码求解的是无向有权图的最短路径,如果想求有向有权图的最短路径,则只需要将无向图的邻接矩阵改为有向图的邻接矩阵即可。

import java.util.Scanner;

public class SSSP
{
	public static void main(String[] args)
	{
		Scanner input = new Scanner(System.in);
		
		System.out.print("请输入图的顶点和边的个数(格式:顶点个数 边个数):");
		int n = input.nextInt(); //顶点的个数
		int m = input.nextInt(); //边的个数
		
		System.out.println();
		
		int[][] a = new int[n + 1][n + 1];
		//初始化邻接矩阵
		for(int i = 0; i < a.length; i++)
		{
			for(int j = 0; j < a.length; j++)
			{
				a[i][j] = -1; //初始化没有边
			}
		}
		
		System.out.println("请输入图的路径长度(格式:起点 终点 长度):");
		//总共m条边
		for(int i = 0; i < m; i++)
		{
			//起点,范围1到n
			int s = input.nextInt();
			//终点,范围1到n
			int e = input.nextInt();
			//长度
			int l = input.nextInt();
			
			if(s >= 1 && s <= n && e >= 1 && e <= n)
			{
				//无向有权图
				a[s][e] = l;
				a[e][s] = l;
			}
		}
		
		System.out.println();
		
		//距离数组
		int[] dist = new int[n+1];
		//前驱节点数组
		int[] prev = new int[n+1];
		
		int v =1 ;//顶点,从1开始
		dijkstra(v, a, dist, prev);
	}
	
	/**
	 * 单源最短路径算法(迪杰斯特拉算法)
	 * @param v 顶点
	 * @param a 邻接矩阵表示图
	 * @param dist 从顶点v到每个点的距离
	 * @param prev 前驱节点数组
	 */
	public static void dijkstra(int v, int[][] a, int[] dist, int[] prev)
	{
		int n = dist.length;
		/**
		 * 顶点从1开始,到n结束,一共n个结点
		 */
		if(v > 0 && v <= n)
		{
			//顶点是否放入的标志
			boolean[] s = new boolean[n];
			
			//初始化
			for(int i = 1; i < n; i++)
			{
				//初始化为 v 到 i 的距离
				dist[i] = a[v][i];
				//初始化顶点未放入
				s[i] = false;
				//v到i无路,i的前驱节点置空
				if(dist[i] == -1)
				{
					prev[i] = 0;
				}
				else
				{
					prev[i] = v;
				}
			}
			
			//v到v的距离是0
			dist[v] = 0;
			//顶点放入
			s[v] = true;
			
			//共扫描n-2次,v到v自己不用扫
			for(int i = 1; i < n - 1; i++)
			{
				int temp = Integer.MAX_VALUE;
				//u为下一个被放入的节点
				int u = v;
				
				//这个for循环为第二步,观测域为v的观测域
				//遍历所有顶点找到下一个距离最短的点
				for(int j = 1; j < n; j++)
				{
					//j未放入,且v到j有路,且v到当前节点路径更小
					if(!s[j] && dist[j] != -1 && dist[j] < temp)
					{
						u = j;
						//temp始终为最小的路径长度
						temp = dist[j];
					}
				}
				
				//将得到的下一节点放入
				s[u] = true;
				
				//这个for循环为第三步,用u更新观测域
				for(int k = 1; k < n; k++)
				{
					if(!s[k] && a[u][k] != -1)
					{
						int newdist=dist[u] + a[u][k];
						if(newdist < dist[k] || dist[k] == -1)
						{
							dist[k] = newdist;
							prev[k] = u;
						}
					}
				}
			}
		}
		
		for(int i = 2; i < n; i++)
		{
			System.out.println(i + "节点的最短距离是:"
				+ dist[i] + ";前驱点是:" + prev[i]);
		}

	}
}

运行结果

j结果示例