【PAT甲级】1056 Mice and Rice (25 分)

题意：

输入两个正整数N和M（<=1000），接着输入两行，每行N个数，第一行为每只老鼠的重量，第二行为每只老鼠出战的顺序。输出它们的名次。（按照出战顺序每M只老鼠分为一组，剩余不足M只为一组，每组只能有一个胜者，其他老鼠排名均为这一轮胜者数量+1）

AAAAAccepted code：

 #define HAVE_STRUCT_TIMESPEC

 #include<bits/stdc++.h>

 using namespace std;

 int n,m;

 int a[],b[];

 int num;

 int ans[];

 int c[],d[];

 pair<int,int>e[];

 int flag;

 bool vis[];

 void contest(){

     int cnt=;

     int round=;

     int no=;

     for(int i=;i<=num;++i){

         c[++cnt]=b[i];

         if(cnt==m||i==num){

             ++round;

             int mx=a[c[]],pos=c[];

             for(int j=;j<=cnt;++j)

                 if(a[c[j]]>mx){

                     mx=a[c[j]];

                     pos=c[j];

                 }

             for(int j=;j<=cnt;++j)

                 if(c[j]==pos)

                     d[++no]=c[j];

                 else

                     vis[c[j]]=;

             cnt=;

         }

     }

     for(int i=;i<=num;++i)

         if(!ans[b[i]]&&vis[b[i]])

             ans[b[i]]=round+;

     if(round==){

         ans[d[]]=;

         flag=;

     }

     for(int i=;i<=no;++i)

         b[i]=d[i];

     num=no;

     no=;

 }

 int main(){

     ios::sync_with_stdio(false);

     cin.tie(NULL);

     cout.tie(NULL);

     cin>>n>>m;

     for(int i=;i<=n;++i)

         cin>>a[i];

     for(int i=;i<=n;++i){

         cin>>b[i];

         ++b[i];

     }

     num=n;

     while(!flag)

         contest();

     cout<<ans[];

     for(int i=;i<=n;++i)

         cout<<" "<<ans[i];

     return ;

 }