力扣OJ 210. 課程表 II

現在你總共有 n 門課需要選,記爲 0 到 n-1。

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

給定課程總量以及它們的先決條件,返回你爲了學完所有課程所安排的學習順序。

可能會有多個正確的順序,你只要返回一種就可以了。如果不可能完成所有課程,返回一個空數組。

示例 1:

輸入: 2, [[1,0]] 
輸出: [0,1]
解釋: 總共有 2 門課程。要學習課程 1,你需要先完成課程 0。因此,正確的課程順序爲 [0,1] 。
示例 2:

輸入: 4, [[1,0],[2,0],[3,1],[3,2]]
輸出: [0,1,2,3] or [0,2,1,3]
解釋: 總共有 4 門課程。要學習課程 3,你應該先完成課程 1 和課程 2。並且課程 1 和課程 2 都應該排在課程 0 之後。
     因此,一個正確的課程順序是 [0,1,2,3] 。另一個正確的排序是 [0,2,1,3] 。
說明:

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

這個問題相當於查找一個循環是否存在於有向圖中。如果存在循環,則不存在拓撲排序,因此不可能選取所有課程進行學習。
通過 DFS 進行拓撲排序 - 一個關於Coursera的精彩視頻教程(21分鐘),介紹拓撲排序的基本概念。
拓撲排序也可以通過 BFS 完成。

 

map<int, int>visitt;//單次訪問標記
map<int, int>flag;//所有訪問標記
map<int, int>inNum;//入度
map<int, int>outNum;//出度

class Solution {
public:
	bool canFinish(vector<vector<int>>& diag,int loc) {
		if (visitt[loc] == 1)return false;
		visitt[loc] = 1, flag[loc] = 1;
		for (int i = 0; i < diag[loc].size(); i++)
		{
			if (!canFinish(diag, diag[loc][i]))return false;
		}
		visitt[loc] = 0;
		return true;
	}
	bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
		vector<int>tmp;
		vector<vector<int>>diag;
		for (int i = 0; i < numCourses; i++)
		{
			diag.insert(diag.end(), tmp);
			visitt[i] = 0, inNum[i] = 0, flag[i] = 0;
		}
		for (int i = 0; i < prerequisites.size(); i++)
		{
			diag[prerequisites[i][0]].insert(diag[prerequisites[i][0]].end(), prerequisites[i][1]);
			inNum[prerequisites[i][1]]++;
		}
		for (int i = 0; i < numCourses; i++)
		{
			if (flag[i])continue;
			if (!canFinish(diag, i))return false;
		}
		return true;
	}
    void findOrder(vector<vector<int>>& diag,int loc,vector<int>&ans)
    {
        if(outNum[loc] || flag[loc])return;
        ans.push_back(loc);
        flag[loc]++;
        for(int i=0;i<diag[loc].size();i++)
        {
            outNum[diag[loc][i]]--;
            findOrder(diag,diag[loc][i],ans);
        }
    }
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        vector<int>tmp;
        if(!canFinish(numCourses,prerequisites))return tmp;
		vector<vector<int>>diag;
		for (int i = 0; i < numCourses; i++)
		{
			diag.insert(diag.end(), tmp);
			visitt[i] = 0, outNum[i] = 0, flag[i] = 0;
		}
		for (int i = 0; i < prerequisites.size(); i++)
		{
			diag[prerequisites[i][1]].insert(diag[prerequisites[i][1]].end(), prerequisites[i][0]);
			outNum[prerequisites[i][0]]++;
		}
        vector<int>ans;
        flag.clear();
        for (int i = 0; i < numCourses; i++)
		{
            findOrder(diag,i,ans);
        }
        return ans;
    }
};

 

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