PAT 甲1056 Mice and Rice 队列

#include<cstdio>
#include<queue>
using namespace std;
queue<int>q;
struct mouse{
    int w;
    int rank;
}mouse[1010];
int main(){
    int NP,NG,order;
    scanf("%d%d",&NP,&NG);
    for(int i=0;i<NP;i++){
        scanf("%d",&mouse[i].w);
    }
    for(int i=0;i<NP;i++){
        scanf("%d",&order);
    	q.push(order);
    }
    //temp记录每一轮多少参赛者 
	int temp=NP,group;
	int k;//记录一个group中最大的编号 
	while(q.size()!=1){
		if(temp%NG==0)
			group=temp/NG;
		else
			group=temp/NG+1;
		for(int i=0;i<group;i++){
			k=q.front();
			for(int j=0;j<NG;j++){
				if(i*NG+j>=temp) break;
				int o=q.front();
				if(mouse[o].w>mouse[k].w)
					k=o;
				mouse[o].rank=group+1;
				q.pop();
			}
			mouse[k].rank-=1;
			q.push(k);
		}
		temp=group;
	}
	mouse[q.front()].rank=1;
	for(int i=0;i<NP;i++)
	{
		if(i!=NP-1)
		printf("%d ",mouse[i].rank);
		else
		printf("%d\n",mouse[i].rank);
	}
}