玖雨y的算法刷题记录day12——图论

玖雨y的算法刷题记录day12——图论

• 题目链接：参加科学大会
• 题目思路：本题求最短路径，这是图论的一个经典问题，我们采用dijkstra算法(在有权图(权值非负数)中求从起点到其他节点的最短路径算法)来求解本题，dijkstra算法的思想和最小生成树的Prim算法很相似，可以分为以下三步：

1. 选源点到哪个节点最近且该节点未被访问过
2. 标记该最近节点为访问过
3. 更新非访问节点到源点的距离（即更新minDist数组）

注意dijkstra算法更新距离的方式，是计算到源点的距离，而非像Prim算法那样计算到生成树的距离，这就是二者之间的不同。

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();

        int[][] grid = new int[n + 1][n + 1];
        for (int i = 0; i <= n; i++) {
            Arrays.fill(grid[i], Integer.MAX_VALUE);
        }

        for (int i = 0; i < m; i++) {
            int p1 = scanner.nextInt();
            int p2 = scanner.nextInt();
            int val = scanner.nextInt();
            grid[p1][p2] = val;
        }

        int start = 1;
        int end = n;

        // 存储从源点到每个节点的最短距离
        int[] minDist = new int[n + 1];
        Arrays.fill(minDist, Integer.MAX_VALUE);

        // 记录顶点是否被访问过
        boolean[] visited = new boolean[n + 1];

		// 因为代码不会对其进行更新，而计算后续距离需要在此基础上计算，所以需要提前将起始点到自身的距离置为0
        minDist[start] = 0;  
		
		// 遍历所有节点
        for (int i = 1; i <= n; i++) { 

            int minVal = Integer.MAX_VALUE;
            int cur = 1;

            // 1、选距离源点最近且未访问过的节点
            for (int v = 1; v <= n; ++v) {
                if (!visited[v] && minDist[v] < minVal) {
                    minVal = minDist[v];
                    cur = v;
                }
            }

			// 2、标记该节点已被访问
            visited[cur] = true;  

            // 3、更新非访问节点到源点的距离（即更新minDist数组）
            for (int v = 1; v <= n; v++) {
                if (!visited[v] && grid[cur][v] != Integer.MAX_VALUE && minDist[cur] + grid[cur][v] < minDist[v]) {
                    minDist[v] = minDist[cur] + grid[cur][v];
                }
            }
        }

        if (minDist[end] == Integer.MAX_VALUE) {
        	// 不能到达终点
            System.out.println(-1); 
        } else {
        	// 到达终点最短路径
            System.out.println(minDist[end]); 
        }
    }
}