【Leetcode】1136. Parallel Courses

题目地址:

https://leetcode.com/problems/parallel-courses/

给定一个课程的先修后修关系(即某个课程要修必须得其所有先修课程全修完才行),问至少要多少个学期能把所有课修完。每个学期修课数量不限制。如果不存在方案,则返回− 1 -11

思路是拓扑排序。这里由于要计算拓扑排序的层数,所以只能用BFS来做。开个队列,先将所有入度为0 00的点入队,然后进行分层遍历。每遍历一层的时候,就将其出边删去,一旦发现出边删去导致其邻接点入度为0 00了,则将这个邻接点入队。BFS的时候累加层数和出队点的个数(出队点的个数就是已经排好序的点的个数)。最后,如果所有的点都出过队了,则说明存在拓扑排序,返回层数;否则返回− 1 -11。代码如下:

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;

public class Solution {
    public int minimumSemesters(int N, int[][] relations) {
    	// 邻接表建图,并且存一下每个点的入度
        int[] indegrees = new int[N];
        List<Integer>[] graph = buildGraph(relations, N, indegrees);
        
        // 先将入度为0的点入队
        Queue<Integer> queue = new ArrayDeque<>();
        for (int i = 0; i < N; i++) {
            if (indegrees[i] == 0) {
                queue.offer(i);
            }
        }
        
        int res = 0;
        while (!queue.isEmpty()) {
            res++;
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                int cur = queue.poll();
                N--;
                if (graph[cur] != null) {
                    for (int next : graph[cur]) {
                        indegrees[next]--;
                        if (indegrees[next] == 0) {
                            queue.offer(next);
                        }
                    }
                }
            }
        }
        
        // 如果课没全修完,说明存在环,返回-1,否则返回学期数
        return N == 0 ? res : -1;
    }
    
    private List<Integer>[] buildGraph(int[][] relations, int N, int[] indegrees) {
        List<Integer>[] graph = (List<Integer>[]) new ArrayList[N];
        for (int[] relation : relations) {
            int from = relation[0] - 1, to = relation[1] - 1;
            if (graph[from] == null) {
                graph[from] = new ArrayList<>();
            }
            
            graph[from].add(to);
            indegrees[to]++;
        }
        
        return graph;
    }
}

时空复杂度O ( V + E ) O(V+E)O(V+E)


版权声明:本文为qq_46105170原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。