pat 甲级 1056 Mice and Rice

#include<iostream>
#include<algorithm>
#include<queue>

using namespace std;
const int N=1010;
struct mice{
    int w,rank;
}mices[N];
int main()
{
    int n,m;
    cin>>n>>m;
    queue<int> q;
    for(int i=0;i<n;i++) cin>>mices[i].w;
    for(int i=0;i<n;i++){
        int index;
        cin>>index;
        q.push(index);
    }
    int temp=n;
    int group;
    while(q.size()>1){
        if(temp%m==0) group=temp/m; //group代表有几组
        else group=temp/m+1;
        for(int i=0;i<group;i++){
            int k=q.front(); 
            for(int j=1;j<=m;j++){
                if(i*m+j>temp) break;//因为每组的最大值会重新放入队列，若超过当前的则退出
                if(mices[q.front()].w>mices[k].w) k=q.front();//k保存该组的最大值
                mices[q.front()].rank=group+1;//因为对于共有group组的老鼠，每组晋级一个，也就是说最终这一轮能晋级的是group个老鼠，那么没有晋级的所有人就是group+1名，就像有4个人晋级下一轮，那么所有没晋级的这一轮就都是并列第5名
                q.pop();//弹出队列
            }
            q.push(k);//将一组的最大值放入队列
        }
        temp=group;//每组的最大值放入队列，该队列的长度变为group。不断重复，直至队列个数为1
    }
    mices[q.front()].rank=1;
    for(int i=0;i<n;i++){
        if(!i) cout<<mices[i].rank;
        else cout<<" "<<mices[i].rank;
    }
    cout<<endl;
    return 0;
}