zoukankan      html  css  js  c++  java
  • 210. Course Schedule II

    package LeetCode_210
    
    import java.util.*
    import kotlin.collections.ArrayList
    
    /**
     * 210. Course Schedule II
     * https://leetcode.com/problems/course-schedule-ii/description/
     *
     * There are a total of n courses you have to take, labeled from 0 to n-1.
    Some courses may have prerequisites, for example to take course 0 you have to first take course 1, which is expressed as a pair: [0,1]
    Given the total number of courses and a list of prerequisite pairs, return the ordering of courses you should take to finish all courses.
    There may be multiple correct orders, you just need to return one of them. If it is impossible to finish all courses, return an empty array.
    
    Example 1:
    Input: 2, [[1,0]]
    Output: [0,1]
    Explanation: There are a total of 2 courses to take. To take course 1 you should have finished
    course 0. So the correct course order is [0,1] .
     * */
    class Solution {
        /**
         * solution: BFS
         * Time complexity:O(n^2), Space complexity:O(n)
         * */
        fun findOrder(numCourses: Int, prerequisites: Array<IntArray>): IntArray {
            //init graph
            val graph = ArrayList<ArrayList<Int>>()
            for (i in 0 until numCourses) {
                graph.add(ArrayList())
            }
            for (item in prerequisites) {
                val start = item[1]
                val end = item[0]
                graph[start].add(end)
            }
    
            //calculate in degree
            val inDegree = Array<Int>(graph.size, { 0 })
            for (item in graph) {
                for (v in item) {
                    inDegree[v]++
                }
            }
    
            val result = IntArray(numCourses)
    
            val queue = LinkedList<Int>()
    
            for (i in 0 until numCourses) {
                if (inDegree[i] == 0) {
                    queue.offer(i)
                }
            }
    
            for (i in 0 until numCourses) {
                if (queue.isEmpty()) {
                    return IntArray(0)
                }
    
                val top = queue.poll()
                result.set(i, top)
    
                for (item in graph[top]) {
                    inDegree[item]--
                    if (inDegree[item] == 0) {
                        queue.offer(item)
                    }
                }
            }
    
            return result
        }
    
    }
  • 相关阅读:
    BZOJ-1034-[ZJOI2008]泡泡堂BNB(贪心)
    BZOJ-2456-mode(思维题)
    POJ-2528-Mayor's posters(线段树+离散化)
    POJ-2352-Stars(树状数组)
    HDU-2688-Rotate(树状数组)
    POJ-1195-Mobile phones(二维树状数组)
    YYHS-NOIP2017Training0921-逆光
    YYHS-鏖战字符串(斜率优化)
    左偏树
    2-sat模板
  • 原文地址:https://www.cnblogs.com/johnnyzhao/p/13216319.html
Copyright © 2011-2022 走看看