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