A1056 Mice and Rice (25 分)

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <queue>
using namespace std;
int N, G;
struct Mice{
	int w, turn, rank;
}mice[1010];
int main(){
	scanf("%d %d", &N, &G);
	for(int i = 0; i < N; i++){
		scanf("%d", &mice[i].w);
		mice[i].turn = 0;
	}
	queue<int> que;
	int no;
	for(int i = 0; i < N; i++){
		scanf("%d", &no);
		que.push(no);
	}
	int turn = 0, n = N;
	while(que.size() > 1){
		int out = 0, maxw = -1, index;
		for(int i = 0; i < G; i++){
			no = que.front();
			if(que.empty() || mice[no].turn > turn) break;
			if(mice[no].turn == turn){
				que.pop(); out++;
				mice[no].rank = ceil(n*1.0/G) + 1;
				if(maxw < mice[no].w){
					maxw = mice[no].w; index = no;
				}
			}
		}
		if(out > 0){
			mice[index].turn++; que.push(index);
		}else{
			turn++; n = ceil(n*1.0/G);
		}		
	}
	no = que.front(); que.pop();
	mice[no].rank = 1;
	printf("%d", mice[0].rank);
	for(int i = 1; i < N; i++){
		printf(" %d", mice[i].rank);
	}	
	return 0;
}