安排课程-LintCode

你需要去上n门九章的课才能获得offer，这些课被标号为 0 到 n-1 。

有一些课程需要“前置课程”，比如如果你要上课程0，你需要先学课程1，我们用一个匹配来表示他们： [0,1]

给你课程的总数量和一些前置课程的需求，返回你为了学完所有课程所安排的学习顺序。

可能会有多个正确的顺序，你只要返回一种就可以了。如果不可能完成所有课程，返回一个空数组。

样例：

给定 n = 2, prerequisites = [[1,0]]

返回 [0,1]

给定 n = 4, prerequisites = [1,0],[2,0],[3,1],[3,2]]

返回 [0,1,2,3] or [0,2,1,3]

思路：

拓扑排序。

#ifndef C616_H
#define C616_H
#include<iostream>
#include<vector>
#include<queue>
#include<utility>
using namespace std;
class Solution {
public:
/*
* @param numCourses: a total of n courses
* @param prerequisites: a list of prerequisite pairs
* @return: the course order
*/
vector<int> findOrder(int numCourses, vector<pair<int, int>> &prerequisites) {
// write your code here
vector<int> res;
if (numCourses <= 0)
return res;
queue<int> que;
vector<vector<int>> graph(numCourses);//构建邻接矩阵存放图
vector<int> indegree(numCourses,0);   //存放每个节点的入度
for (auto c : prerequisites)
{
indegree[c.second]++;
graph[c.first].push_back(c.second);
}
//将入度为0的节点入队
for (int i = 0; i < numCourses; ++i)
{
if (indegree[i] == 0)
que.push(i);
}
//将队头节点插入到res的开始位置
while (!que.empty())
{
int p = que.front();
res.insert(res.begin(),p);
que.pop();
//遍历队头节点的邻接节点
for (int j = 0; j < graph[p].size(); ++j)
{
if (--indegree[graph[p][j]]==0)
que.push(graph[p][j]);
}
}
if (res.size() < numCourses)
res.clear();
return res;
}
};
#endif