课程表
你这个学期必须选修 numCourses 门课程,记为 0 到 numCourses - 1 。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出,其中 prerequisites[i] = [ai, bi] ,表示如果要学习课程 ai 则 必须 先学习课程 bi 。
例如,先修课程对 [0, 1] 表示:想要学习课程 0 ,你需要先完成课程 1 。 请你判断是否可能完成所有课程的学习?如果可以,返回 true ;否则,返回 false 。
示例 1:
codeType
输入:numCourses = 2, prerequisites = [[1,0]]
输出:true
解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。
示例 2:
codeType
输入:numCourses = 2, prerequisites = [[1,0],[0,1]]
输出:false
解释:总共有 2 门课程。学习课程 1 之前,你需要先完成课程 0 ;并且学习课程 0 之前,你还应先完成课程 1 。这是不可能的。
提示:
- 1 <= numCourses <= 2000
- 0 <= prerequisites.length <= 5000
- prerequisites[i].length == 2
- 0 <= ai, bi < numCourses
- prerequisites[i] 中的所有课程对 互不相同
解法
ts
function canFinish(numCourses: number, prerequisites: number[][]): boolean {
// 这种数据可以用有向无环图记录(拓扑排序)
// 从入度为0度课开始学习,学习以后依赖其的课的入度就-1
// 依次学习入度为0的课,最后如果还剩入度不为0的节点,那么就说明学不完(其实就是图存在环)
// 构建图,临接表
const map = new Map<number, // 课程
{
inDegree: number, // 该课程的入度
deps: number[], // 后续课程 比如A是B的前置课程,则A的该列表中有B
}>()
prerequisites.forEach(item => {
const [ cur, pre ] = item // 当前课程 前置课程
if(map.has(pre)){
map.get(pre).deps.push(cur) // cur依赖于pre课程
}else{
map.set(pre, {
inDegree: 0,
deps: [ cur ]
})
}
if(map.has(cur)){
map.get(cur).inDegree++ // cur课程的入度++
}else{
map.set(cur, {
inDegree: 1,
deps: []
})
}
})
// 从入度为0的节点开始删除
const q = []
for(const [ cur, { inDegree } ] of map.entries()){
if(inDegree === 0){ // 入度为0的节点先入队
q.push(cur)
}
}
while(q.length !== 0){
const curCourse = q.shift()
// 所有以该课程为前置课程的课程入度--
map.get(curCourse).deps.forEach(course => {
if(--map.get(course).inDegree === 0){ // 入度-- 后如果为0要继续入队
q.push(course)
}
})
map.delete(curCourse) // 删除该课程
}
return map.size === 0
};