逃生
Problem Description
糟糕的事情发生啦,现在大家都忙着逃命。但是逃命的通道很窄,大家只能排成一行。
现在有n个人,从1标号到n。同时有一些奇怪的约束条件,每个都形如:a必须在b之前。
同时,社会是不平等的,这些人有的穷有的富。1号最富,2号第二富,以此类推。有钱人就贿赂负责人,所以他们有一些好处。
负责人现在可以安排大家排队的顺序,由于收了好处,所以他要让1号尽量靠前,如果此时还有多种情况,就再让2号尽量靠前,如果还有多种情况,就让3号尽量靠前,以此类推。
那么你就要安排大家的顺序。我们保证一定有解
思路
举个例子如图:
如果你用优先队列拓扑排序得到的是:3 5 6 4 1 7 8 9 2 0
但是正确答案为 6 4 1 3 9 2 5 7 8 0 这样使得小的(1)尽量在前面。
这里我们可以得到 前面的小的不一定排在前面,但是有一点后面大的一定排在后面。
我们看 6和3不一定3排在前面,因为6后面连了一个更小的数字1能使得6更往前排。
在看 2和 8,8一定排在后面,因为8后面已经没有东西能使它更往前排(除了0)。
所以最后我们的做法就是 建立一个反图,跑一边字典序最大的拓扑排序,最后再把这个排序倒过来就是答案了。
AC Code
#include<bits/stdc++.h>
using namespace std;
const int maxn = 3e4+10;
vector<int>edge[maxn],ans;
priority_queue<int>q;
int in[maxn];
int n,m;
void topo() {
for(int i=1; i<=n; i++) {
if(in[i] == 0) q.push(i);
}
while(!q.empty()) {
int p = q.top();
q.pop();
ans.push_back(p);
for(int i=0; i<edge[p].size(); i++) {
int k = edge[p][i];
in[k]--;
if(in[k] == 0) q.push(k);
}
}
}
void init() {
for(int i=0; i<=n; i++)
edge[i].clear();
while(!q.empty()) q.pop();
ans.clear();
}
int main() {
int T,a,b;
scanf("%d",&T);
while(T--) {
init();
scanf("%d%d",&n,&m);
for(int i=0; i<m; i++) { //反向建图
scanf("%d%d",&a,&b);
in[a]++;
edge[b].push_back(a);
}
topo();
for(int i=ans.size()-1; i>0; i--)
printf("%d ",ans[i]);
printf("%d\n",ans[0]);
}
return 0;
}