Leetcode C++《熱題 Hot 100-65》207. 課程表

Leetcode C++《熱題 Hot 100-65》207. 課程表

1. 題目

你這個學期必須選修 numCourse 門課程,記爲 0 到 numCourse-1 。

在選修某些課程之前需要一些先修課程。 例如,想要學習課程 0 ,你需要先完成課程 1 ,我們用一個匹配來表示他們:[0,1]

給定課程總量以及它們的先決條件,請你判斷是否可能完成所有課程的學習?

示例 1:

輸入: 2, [[1,0]]
輸出: true
解釋: 總共有 2 門課程。學習課程 1 之前,你需要完成課程 0。所以這是可能的。
示例 2:

輸入: 2, [[1,0],[0,1]]
輸出: false
解釋: 總共有 2 門課程。學習課程 1 之前,你需要先完成​課程 0;並且學習課程 0 之前,你還應先完成課程 1。這是不可能的。

提示:

輸入的先決條件是由 邊緣列表 表示的圖形,而不是 鄰接矩陣 。詳情請參見圖的表示法。
你可以假定輸入的先決條件中沒有重複的邊。
1 <= numCourses <= 10^5

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/course-schedule
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

2. 思路

  • 拓撲排序,用出入度方法容易理解hh,熟悉的配方。[0, 1], 先學習1才能學習0,那麼0的入度爲++;學習1之後要把它學了之後能學的課程入度都–,需要一個map
  • 如果不能完美的進行拓撲排序,返回false

3. 代碼

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        //拓撲排序,用出入度方法容易理解hh,熟悉的配方
        // [0, 1], 先學習1才能學習0,那麼0的入度爲++;學習1之後要把它學了之後能學的課程入度都--,需要一個map
        map<int, int> degrees;
        map<int, vector<int>> relations;
        queue<int> temp;
        map<int, int>:: iterator it_degree;
        map<int, vector<int>>:: iterator it_relation;

        for (int i = 0; i < prerequisites.size(); i++) {
            vector<int> relation = prerequisites[i];
            it_degree = degrees.find(relation[0]);
            if (it_degree == degrees.end()) {
                degrees.insert(make_pair(relation[0], 1));
            } else {
                degrees[relation[0]]++;
            }
            it_relation = relations.find(relation[1]);
            if (it_relation == relations.end()) {
                vector<int> childrens;
                childrens.push_back(relation[0]);
                relations.insert(make_pair(relation[1], childrens));
            } else {
                relations[relation[1]].push_back(relation[0]);
            }
        }

        //找到入度爲0的點,放進temp
        vector<int> find_zero;
        for (int i = 0; i < numCourses; i++) {
            it_degree = degrees.find(i);
            if (it_degree == degrees.end())
                temp.push(i);
        }
        /*for (it_degree = degrees.begin(); it_degree != degrees.end(); it_degree++) {
            cout << it_degree->first << " " << it_degree->second << endl;
        }*/

        int nums = 0;
        while (!temp.empty()) {
            int one_lesson = temp.front();
            temp.pop();
            nums++;
            it_relation = relations.find(one_lesson);
            if (it_relation != relations.end()) {
                vector<int> childrens_lesson = relations[one_lesson];
                for (int i = 0; i < childrens_lesson.size(); i++) {
                    degrees[childrens_lesson[i]]--;
                    if (degrees[childrens_lesson[i]] == 0) {
                        temp.push(childrens_lesson[i]);
                    }
                    if (degrees[childrens_lesson[i]] < 0) {
                       return false;
                    }
                }
            }
        }
        if (nums == numCourses)
            return true;
        return false;

    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章