HRBUST-1631.技能修炼
Description
寒假第三次周赛强势袭来,首先祝愿大能依靠自己所学的取得一个好成绩。学习之余,我们在这里说一个有关游戏的问题,你来解决一下。一般的游戏中人物技能在修炼时有个特点,就是后面的技能一般都要修炼了前面的某些个才可以修炼。比如10级有一个技能火弹术,20级有一个火焰术,那么此时假如你没修炼10级的火弹,你就不能直接修20级的火焰术,就是这样的道理。现在的问题是,给定一些个技能间的依赖关系,你来写程序来判断该以什么样的顺序来修炼这些技能。为了简化问题,我们用数代表不同的技能,一个数代表一种技能,输入形式N1 N2则代表N1是N2的前置技能。
Input
本题有多组测试数据,每组中第一行为两个数N(1 <= N <= 500)和M(1 <= M <= N*(N-1)/2),其中N表示技能总数,接下来有M行。每行是代表两个技能的数字N1和N2,表示N1是N2的前置技能。
Output
对每组测试数据,输出确定下来的技能修炼顺序,要求对于输入的每组数据,输出一行,一行中各技能之间以一个空格间隔(行尾不该有空格而是换行符)。
注意,对于排序的结果可能不是唯一的,这时要求先输出代表技能的数较小的。
Sample Input
4 3
1 2
2 3
4 3
Sample Output
1 2 4 3
Hint
对于上面的样例中2和4都是3的前置技能,但是数字2较小,因此先输出2。
拓扑排序
因为题目要求按基数小的优先输出,所以可以用二叉堆来保存入度为0的点。
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
int n,m;
vector<vector<int>> vec;
vector<int> a;
vector<int> deg; //记录每个点的入度
priority_queue<int,vector<int>,greater<int>> q; // 保存入度为0的点
void toporder(){
for(int i = 1;i <= n;i++){
if(deg[i] == 0) q.push(i);
}
while(!q.empty()){
int x = q.top();
q.pop();
a.push_back(x);
for(int i = 0;i < (int)vec[x].size();i++){
int y = vec[x][i];
deg[y]--;
if(deg[y] == 0) q.push(y);
}
}
}
int main(){
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
std::cout.tie(nullptr);
cin >> n >> m;
vec.resize(n+1);
deg.resize(n+1);
for(int i = 0;i < m;i++){ // 构建邻接表
int x,y;
cin >> x >> y;
vec[x].push_back(y);
deg[y]++;
}
toporder();
for(int i = 0;i < (int)a.size();i++){
cout << a[i] <<" ";
}
cout << endl;
return 0;
}