week8 作业B 猫猫向前冲 kahn

思路分析

众所周知， TT 是一位重度爱猫人士，他有一只神奇的魔法猫。

有一天，TT 在 B 站上观看猫猫的比赛。一共有 N 只猫猫，编号依次为1，2，3，…，N进行比赛。比赛结束后，Up 主会为所有的猫猫从前到后依次排名并发放爱吃的小鱼干。不幸的是，此时 TT 的电子设备遭到了宇宙射线的降智打击，一下子都连不上网了，自然也看不到最后的颁奖典礼。

不幸中的万幸，TT 的魔法猫将每场比赛的结果都记录了下来，现在他想编程序确定字典序最小的名次序列，请你帮帮他。

样例

Input
4 3
1 2
2 3
4 3
Output
1 2 4 3

思路分析

可以把猫猫的输赢关系转化为有向图，根据题意可知是要求这个有向图中字典序最小的拓跋序列。（kahn算法）

void kahn()
{
    priority_queue<int> q;
 for(int i=1;i<=n;i++)
 {
  if(in_deg[i]==0)
  {
   q.push(-i);
   //cout<<"in "<<i<<endl;
  }
 } 
 vector<int> ans;
 while(!q.empty())
 {
  int u=-q.top();q.pop();
  ans.push_back(u);
  //cout<<u<<endl;
  for (int i= head[u];~i;i= edge[i].next) 
  {
   int v=edge[i].v;
   in_deg[v]--;
   if(in_deg[v]==0)//入度为0
   {
    q.push(-v);
    //cout<<"in "<<v<<endl;
   } 
     }
 }
}

要求字典序最小，在kahn算法中可以使用优先队列，每次入队的是当前点的负值，那么队首元素的负值就是当前入度为0且字典序最小的点。

AC代码

#include<iostream>
#include<queue>
#include<cstring>
#include<cmath>
const int inf=1e9;
using namespace std;
int head[510],in_deg[510];
struct Edge
{
 int v,next;//相邻的点，权值，下条边的序号 
} ;
int tot,n,m; 
Edge edge[200000];
void init()
{
 //初始化 
 tot=0;
 memset(head,-1,sizeof(head));
 memset(in_deg,0,sizeof(in_deg));
 } 
void add(int u,int v)
{
 edge[tot].v=v;
 edge[tot].next=head[u];
 head[u]=tot;
 tot++;
}
void kahn()
{
    priority_queue<int> q;
 for(int i=1;i<=n;i++)
 {
  if(in_deg[i]==0)
  {
   q.push(-i);
   //cout<<"in "<<i<<endl;
  }
 } 
 vector<int> ans;
 while(!q.empty())
 {
  int u=-q.top();q.pop();
  ans.push_back(u);
  //cout<<u<<endl;
  for (int i= head[u];~i;i= edge[i].next) 
  {
   int v=edge[i].v;
   in_deg[v]--;
   if(in_deg[v]==0)//入度为0
   {
    q.push(-v);
    //cout<<"in "<<v<<endl;
   } 
     }
 }
 for(int i=0;i<n-1;i++)
 {
     cout<<ans[i]<<" ";
 }
 cout<<ans[n-1]<<endl;
}
int main()
{
    while(cin>>n>>m)
    {
     init();
     int p1,p2;
     for(int i=0;i<m;i++)
     {
      cin>>p1>>p2;
      add(p1,p2);
      in_deg[p2]++;
  }
  kahn();
 }
 return 0;
}