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;
}