PAT (Advanced Level) Practice A1056 Mice and Rice （25 分）(C++)（甲级）（队列思想）

原题链接

#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;//没有进入下一轮的排名等于进入下一轮排名的个数+1
        }
    }
    Rank[Q[0]] = 1;//最后一轮的那个排名第一
    printf("%d", Rank[0]);
    for(int i=1; i<Np; i++) printf(" %d", Rank[i]);
    return 0;
}