原题链接
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int MAX = 1010;
int Np, Ng;
int W[MAX] = {0}, Order[MAX] = {0}, Rank[MAX] = {0};
int Q[MAX] = {0}, cnt = 0;
int main()
{
scanf("%d %d", &Np, &Ng);
for(int i=0; i<Np; i++) scanf("%d", &W[i]);
for(int i=0; i<Np; i++)
{
scanf("%d", &Order[i]);
Q[cnt++] = Order[i];
}
while(cnt > 1)
{
int temp = cnt;
cnt = 0;
int visited[MAX] = {0};
for(int i=0; i<temp; i+=Ng)
{
int tag = -1, MAX = -1;
for(int j=i; j<i+Ng && j<temp; j++)
{
if(W[Q[j]] > MAX)
{
MAX = W[Q[j]];
tag = Q[j];
}
}
Q[cnt++] = tag;
visited[tag] = 1;
}
for(int i=0; i<Np; i++)
{
if(!visited[i] && !Rank[i]) Rank[i] = cnt+1;
}
}
Rank[Q[0]] = 1;
printf("%d", Rank[0]);
for(int i=1; i<Np; i++) printf(" %d", Rank[i]);
return 0;
}