数据结构算法——1102. 任务调度问题
题目
思路
拓扑排序问题 用队列进行储存 把入度为0的点先存进去
然后利用vector来储存每个点对应的后置课程(出度 并用一个数组记录每个课程的前置课程数量
在出栈的时候把该前置课程对应的后置课程数组储存的数字-1(也就是入度-1) 在这个过程中数组储存的数变成0的话,入栈,如此往复
代码
#include<iostream> #include<vector> #include<queue> using namespace std; vector<int> edge[200]; int pre_course[200]; int main() { int n; cin >> n; queue<int> q; for(int i = 1; i <= n; i++) { int m; cin >> m; pre_course[i] = m; if(m == 0) q.push(i); for(int j = 0; j < m; j++) { int pre; cin >> pre; edge[pre].push_back(i); //该前置课程对应的后续解锁课程 } } while(!q.empty()) { int pre = q.front(); q.pop(); for(auto data : edge[pre]) { pre_course[data]--;//所有的后续课程完成一个前置课程 if(!pre_course[data])//完成所有前置课程 q.push(data); } } for(int i = 1; i <= n; i++) { if(pre_course[i]) { cout << 0 << endl; return 0; } } cout << 1 << endl; }
下一篇:
【动态规划】插入乘号问题