刷题使我快乐,满脸开心.jpg
- 来源:力扣(LeetCode)
- 链接:https://leetcode.cn/problems/course-schedule-iv
- 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目
你总共需要上 numCourses
门课,课程编号依次为 0
到 numCourses-1
。你会得到一个数组 prerequisite
,其中 prerequisites[i] = [ai, bi]
表示如果你想选 bi
课程,你 必须
先选 ai
课程。
有的课会有直接的先修课程,比如如果想上课程 1
,你必须先上课程 0
,那么会以 [0,1]
数对的形式给出先修课程数对。
先决条件也可以是 间接
的。如果课程 a
是课程 b
的先决条件,课程 b
是课程 c
的先决条件,那么课程 a
就是课程 c
的先决条件。
你也得到一个数组 queries
,其中 queries[j] = [uj, vj]
。对于第 j
个查询,您应该回答课程 uj
是否是课程 vj
的先决条件。
返回一个布尔数组 answer
,其中 answer[j]
是第 j
个查询的答案。
示例 1:
输入:numCourses = 2, prerequisites = [[1,0]], queries = [[0,1],[1,0]]
输出:[false,true]
解释:课程 0 不是课程 1 的先修课程,但课程 1 是课程 0 的先修课程。
示例 2:
输入:numCourses = 2, prerequisites = [], queries = [[1,0],[0,1]]
输出:[false,false]
解释:没有先修课程对,所以每门课程之间是独立的。
示例 3:
输入:numCourses = 3, prerequisites = [[1,2],[1,0],[2,0]], queries = [[1,0],[1,2]]
输出:[true,true]
提示:
2 <= numCourses <= 100
0 <= prerequisites.length <= (numCourses * (numCourses - 1) / 2)
prerequisites[i].length == 2
0 <= ai, bi <= n - 1
ai != bi
- 每一对
[ai, bi]
都 不同 - 先修课程图中没有环。
1 <= queries.length <= 104
0 <= ui, vi <= n - 1
ui != vi
思路
看题解才想起来自己的思路就是学过的弗洛伊德(Floyd)算法,总之就是先按矩阵的模型列出全部路径可能,点亮连通的路径,然后只要看是不是通路即可了
代码
func checkIfPrerequisite(numCourses int, prerequisites [][]int, queries [][]int) []bool {
routeMatrix := make([][]bool, numCourses)
for i := range routeMatrix {
routeMatrix[i] = make([]bool, numCourses)
}
for _, prerequisity := range prerequisites {
routeMatrix[prerequisity[1]][prerequisity[0]] = true
}
for k := 0; k < numCourses; k++ {
for i := 0; i < numCourses; i++ {
for j := 0; j < numCourses; j++ {
if !routeMatrix[i][j] {
routeMatrix[i][j] = routeMatrix[i][k] && routeMatrix[k][j]
}
}
}
}
res := make([]bool, 0)
for _, query := range queries {
res = append(res, routeMatrix[query[1]][query[0]])
}
return res
}