Leetcode+Java+图论+最小生成树&拓扑排序

53.寻宝

题目描述

在世界的某个区域,有一些分散的神秘岛屿,每个岛屿上都有一种珍稀的资源或者宝藏。国王打算在这些岛屿上建公路,方便运输。

不同岛屿之间,路途距离不同,国王希望你可以规划建公路的方案,如何可以以最短的总公路距离将 所有岛屿联通起来(注意:这是一个无向图)。

给定一张地图,其中包括了所有的岛屿,以及它们之间的距离。以最小化公路建设长度,确保可以链接到所有岛屿。

输入描述

第一行包含两个整数V 和 E,V代表顶点数,E代表边数 。顶点编号是从1到V。例如:V=2,一个有两个顶点,分别是1和2。

接下来共有 E 行,每行三个整数 v1,v2 和 val,v1 和 v2 为边的起点和终点,val代表边的权值。

输出描述

输出联通所有岛屿的最小路径总距离

输入示例
复制代码
7 11
1 2 1
1 3 1
1 5 2
2 6 1
2 4 2
2 3 2
3 4 1
4 5 1
5 6 2
5 7 1
6 7 1
输出示例
复制代码
6
提示信息

数据范围:

2 <= V <= 10000;

1 <= E <= 100000;

0 <= val <= 10000;

如下图,可见将所有的顶点都访问一遍,总距离最低是6.

原理

Prim算法核心

从任意节点开始,逐步"生长"最小生成树

text

复制代码
1. 初始:选节点1,minDist[1]=0,其他=∞
2. 迭代V-1次:
   a. 选未访问节点中minDist最小的cur,加入MST
   b. 用cur更新所有未访问节点的距离
3. 总权 = Σ(minDist[2..V])

代码

java 复制代码
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int V = scanner.nextInt(), E = scanner.nextInt();
        
        // minDist[i]:节点i到已选节点的最近距离,初始∞
        int[] minDist = new int[V + 1];
        Arrays.fill(minDist, 100001);
        
        // visited[i]:节点i是否已加入生成树
        boolean[] visited = new boolean[V + 1];
        
        // 邻接矩阵:grid[i][j] = i到j的边权,无边=∞
        int[][] grid = new int[V + 1][V + 1];
        for (int i = 1; i <= V; i++) Arrays.fill(grid[i], 100001);
        
        // 构建无向图邻接矩阵
        for (int i = 0; i < E; i++) {
            int u = scanner.nextInt(), v = scanner.nextInt(), k = scanner.nextInt();
            grid[u][v] = k;  // u→v
            grid[v][u] = k;  // v→u(无向图)
        }
        
        // Prim算法:V-1次迭代构建MST
        for (int i = 1; i < V; i++) {
            // Step1:找未访问节点中minDist最小的cur
            int cur = -1, minVal = Integer.MAX_VALUE;
            for (int j = 1; j <= V; j++) {
                if (!visited[j] && minDist[j] < minVal) {
                    minVal = minDist[j];
                    cur = j;
                }
            }
            visited[cur] = true;  // 加入生成树
            
            // Step2:用cur更新所有未访问节点的距离
            for (int j = 1; j <= V; j++) {
                if (!visited[j] && grid[cur][j] < minDist[j]) {
                    minDist[j] = grid[cur][j];  // 松弛操作
                }
            }
        }
        
        // 计算MST总权:除起点外所有节点的minDist之和
        int result = 0;
        for (int i = 2; i <= V; i++) {
            result += minDist[i];
        }
        System.out.print(result);
    }
}

原理

Kruskal算法(版本1)

核心贪心 + 并查集

text

复制代码
1. 将所有边按权重从小到大排序
2. 从最小边开始依次添加:
   - 如果两端点不在同一连通分量 → 添加这条边
   - 否则 → 跳过(形成环)
3. 直到连通V-1条边,得到最小生成树

代码

java 复制代码
import java.util.*;

class Edge {
    int l, r, val;  // 起点、终点、权重
    Edge(int l, int r, int val) { this.l = l; this.r = r; this.val = val; }
}

public class Main {
    private static int n = 10001;
    private static int[] father = new int[n];
    
    // 初始化并查集
    public static void init() { for (int i = 0; i < n; i++) father[i] = i; }
    
    // 路径压缩查找
    public static int find(int u) {
        return u == father[u] ? u : (father[u] = find(father[u]));
    }
    
    // 合并集合
    public static void join(int u, int v) {
        u = find(u); v = find(v);
        if (u == v) return;
        father[u] = v;  // 简单合并
    }
    
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int V = scanner.nextInt(), E = scanner.nextInt();
        
        // 存储所有边
        List<Edge> edges = new ArrayList<>();
        for (int i = 0; i < E; i++) {
            int u = scanner.nextInt(), v = scanner.nextInt(), k = scanner.nextInt();
            edges.add(new Edge(u, v, k));
        }
        
        // 按权重排序(核心!)
        edges.sort(Comparator.comparing(edge -> edge.val));
        
        init();  // 初始化并查集
        int result = 0;
        
        // Kruskal算法:从小到大加边
        for (Edge edge : edges) {
            if (find(edge.l) != find(edge.r)) {  // 不同集合才加
                result += edge.val;              // 累加权重
                join(edge.l, edge.r);            // 合并集合
            }
        }
        System.out.print(result);
    }
}

117.软件构建

题目描述

某个大型软件项目的构建系统拥有 N 个文件,文件编号从 0 到 N - 1,在这些文件中,某些文件依赖于其他文件的内容,这意味着如果文件 A 依赖于文件 B,则必须在处理文件 A 之前处理文件 B (0 <= A, B <= N - 1)。请编写一个算法,用于确定文件处理的顺序。

输入描述

第一行输入两个正整数 N, M。表示 N 个文件之间拥有 M 条依赖关系。

后续 M 行,每行两个正整数 S 和 T,表示 T 文件依赖于 S 文件。

输出描述

输出共一行,如果能处理成功,则输出文件顺序,用空格隔开。

如果不能成功处理(相互依赖),则输出 -1。

输入示例
复制代码
5 4
0 1
0 2
1 3
2 4
输出示例
复制代码
0 1 2 3 4
提示信息

文件依赖关系如下:

所以,文件处理的顺序除了示例中的顺序,还存在

0 2 4 1 3

0 2 1 3 4

等等合法的顺序。

数据范围:

0 <= N <= 10 ^ 5

1 <= M <= 10 ^ 9

每行末尾无空格。

原理

Kahn算法(拓扑排序)

核心入度为0 → 处理 → 减少依赖 → 重复

text

复制代码
1. 初始:找所有入度=0的文件入队
2. 循环:
   a. 取出队首文件cur,加入结果
   b. cur处理完 → 减少其所有依赖文件的入度
   c. 新入度=0的文件入队
3. 结果长度==N → 成功;否则 → 存在环

代码

java 复制代码
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();  // 文件数
        int M = scanner.nextInt();  // 依赖关系数
        
        // 入度数组:inDegree[i] = 文件i的依赖数
        int[] inDegree = new int[N];
        
        // 邻接表:list[i] = 文件i依赖的文件列表
        List<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            list.add(new ArrayList<>());
        }
        
        // 构建图 + 计算入度
        for (int i = 0; i < M; i++) {
            int s = scanner.nextInt();  // s → t(t依赖s)
            int t = scanner.nextInt();
            list.get(s).add(t);         // s依赖t
            inDegree[t]++;              // t入度+1
        }
        
        // 标记数组:防止重复入队
        boolean[] flag = new boolean[N];
        
        // 队列:存储入度为0的节点(无依赖,可先处理)
        Queue<Integer> queue = new ArrayDeque<>();
        for (int i = 0; i < N; i++) {
            if (inDegree[i] == 0) {
                queue.add(i);
                flag[i] = true;
            }
        }
        
        // 结果拓扑排序
        List<Integer> result = new ArrayList<>();
        
        // Kahn算法:拓扑排序
        while (!queue.isEmpty()) {
            int cur = queue.poll();     // 取出当前文件
            result.add(cur);            // 加入结果
            
            // 处理cur的所有依赖文件,入度-1
            for (int next : list.get(cur)) {
                inDegree[next]--;
            }
            
            // 重新扫描找新入度为0的节点
            for (int i = 0; i < N; i++) {
                if (!flag[i] && inDegree[i] == 0) {
                    queue.add(i);
                    flag[i] = true;
                }
            }
        }
        
        // 判断是否所有文件都处理完
        if (result.size() == N) {
            // 输出拓扑顺序
            for (int i = 0; i < N; i++) {
                if (i == 0) System.out.print(result.get(i));
                else System.out.print(" " + result.get(i));
            }
        } else {
            System.out.print(-1);  // 存在环,无法排序
        }
    }
}
相关推荐
YSRM3 小时前
Leetcode+Java+图论+并查集
算法·leetcode·图论
小白杨树树4 小时前
【C++】力扣hot100错误总结
c++·leetcode·c#
沐浴露z4 小时前
【JVM】详解 Class类文件的结构
java·jvm·class
桦说编程4 小时前
Java并发编程:两种控制并发度的实现方法及其比较
java·后端
杯莫停丶4 小时前
设计模式之:单例模式
java·单例模式·设计模式
消失的旧时光-19434 小时前
@JvmStatic 的作用
java·开发语言·kotlin
火锅机器5 小时前
java 8 lambda表达式对list进行分组
java·开发语言·list
我是华为OD~HR~栗栗呀5 小时前
华为od-22届考研-测试面经
java·c++·python·功能测试·华为od·华为·面试