1056 Mice and Rice

#include <cstdio>
#include <queue>
using namespace std;

struct Mouse{
    int w;
    int r;
}mouse[1005];

int main()
{
    //freopen("pat.txt","r",stdin);
    int n,step;
    scanf("%d%d",&n,&step);
    int order;
    queue<int> q;
    for(int i=0;i<n;i++)
        scanf("%d",&mouse[i].w);
    for(int i=0;i<n;i++){
        scanf("%d",&order);
        q.push(order);
    }
    int temp=n,group;//temp为当前这一轮参加的老鼠个数，初始化为n；group为组数
    while(q.size()!=1){
        //计算这一轮的分组
        group=(temp%step==0 ? temp/step : temp/step+1);
        //遍历每一组，找出该组的老鼠质量的最大值
        for(int i=1;i<=group;i++){
            int maxIdx=q.front();//记录该组质量最大的下标
            for(int j=1;j<=step;j++){
　　　　　　　　　　//用于判断最后一组不满step个的情况
                if((i-1)*step+j>temp) break;
                if(mouse[q.front()].w > mouse[maxIdx].w)
                    maxIdx=q.front();
                mouse[q.front()].r=group+1;//关键，规律
                q.pop();
            }
            //maxIdx即这一组的胜利者，push进队列，进入下一轮的比赛
            q.push(maxIdx);
        }
        temp=group;//下一轮参加比赛的老鼠个数就是这一轮的组数
    }
    mouse[q.front()].r=1;
    printf("%d",mouse[0].r);
    for(int i=1;i<n;i++)
        printf(" %d",mouse[i].r);
    return 0;
}