HDU-4857 孰先孰后
题目大意:
有 n 个人,编号为 1 到 n ;同时有 k 个规定的次序。要求:
- 首先要满足给定的次序。
- 其次编号小的要在满足次序情况下尽可能前。
情况下这 n 个人的排序。
解题思路:
开始的时候我直接写了拓扑排序,然后将入度为 0 的点放到从小到大排的优先队列里面 ,依次取出来更新其他点入度并存入结果数组,但在输入1 1 3 1 3这个数据后的输出为2 3 1 ,而正确答案应为 3 1 2. 如果按照这个思路写的程序, 没有办法保证满足第二个要求。
正确思路应该为 : 进行拓扑逆排序,同时将入度为 0 的点存入从大到小排的优先队列 , 然后依次取出并更新其他点入度同时存入结果数组 , 输出时将结果数组逆输出 , 即得到想要的拓扑排序结果。
AC代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#define rep(i,l,p) for(int i=l;i<=p;i++)
#define rush() int T; scanf("%d",&T); while(T--)
using namespace std;
const int maxN = 30005;
const int maxM = 100005;
int n,m,ind[maxN];
int order[maxN],p;
vector<int> G[maxN];
priority_queue<int> pque;
void input(){
memset(ind,0,sizeof(ind));
p = 0;
while( !pque.empty() ) pque.pop();
rep(i,0,n){
if ( G[i].size() ) G[i].clear();
}
scanf("%d%d",&n,&m);
int u,v;
rep(i,1,m){
scanf("%d%d",&u,&v);
G[v].push_back(u);
ind[u]++;
}
rep(i,1,n){
if ( !ind[i] ){
pque.push(i);
}
}
return;
}
void solve(){
while( !pque.empty() ){
int u = pque.top();
pque.pop();
order[++p] = u;
for (int i=0;i<G[u].size();i++){
int v = G[u][i];
ind[v]--;
if ( !ind[v] ){
pque.push(v);
}
}
}
return;
}
void output(){
for(int i = p;i>1;i--) printf("%d ", order[i]);
printf("%d\n", order[1]);
return;
}
int main(){
//freopen("in.txt","r",stdin);
rush(){
input();
solve();
output();
}
return 0;
}