这道题是一道很经典的拓扑排序问题
拓扑排序过程: 找到第一个入度为0的顶点,在图中去掉这个顶点,找下一个入度为0的顶点,直至图中没有顶点或者找不到入度为0的顶点为止。
变量:建立邻接链表. 一个统计结点入度的数组, 一个用于保存入度为0的队列.
算法思想:
- 根据输入条件初始化邻接链表和入度数组.
- 选定入度为0的顶点进队列.
- 进入循环,循环终止条件为队列为空
- 循环的动作是弹出队列头,将队列头作为弧起点的弧去掉,弧终点对应的入度–.如果入度为0,加入队列.
代码
//拓扑排序的一个求解问题。
//拓扑排序过程: 建立邻接链表. 一个统计结点入度的数组, 一个用于保存入度为0的队列.
//算法思想,1.根据输入条件初始化邻接链表和入度数组.
// 2.选定入度为0的顶点进队列.
// 3.进入循环,循环终止条件为队列为空
// 4.循环的动作是弹出队列头,将队列头作为弧起点的弧去掉,弧终点对应的入度--.如果入度为0,加入队列.
#define max 10000
class Solution {
public:
vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
vector<int> ret;
vector<int> edge[max];
queue<int> q;
int inDegree[max];
for(int i=0;i<max;i++){
edge[i].clear();
inDegree[i]=0;
}
//初始化邻接链表和入度数组.
for(int i=0;i<prerequisites.size();i++){
//[i][1]是弧的起点,[i][0]是弧的终点
edge[prerequisites[i][1]].push_back(prerequisites[i][0]);
//终点的入度++
inDegree[prerequisites[i][0]]++;
}
//选定入度为0的顶点入队列.
for(int i=0;i<numCourses;i++){
if(inDegree[i]==0){
q.push(i);
}
}
//进入循环.
while(!q.empty()){
int cur=q.front();
q.pop();
ret.push_back(cur);
//将队列头作为弧起点的弧去掉,弧终点对应的入度--,如果入度为0,加入队列.
for(int i=0;i<edge[cur].size();i++){
int tail=edge[cur][i];
inDegree[tail]--;
if(inDegree[tail]==0){
q.push(tail);
}
}
}
if(ret.size()==numCourses)
return ret;
else return {};
}
};