PAT (Advanced Level) 1056. Mice and Rice (25)

简单模拟。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<map>
#include<queue>
#include<string>
#include<stack>
#include<vector>
using namespace std;

const int maxn=+;
struct X
{
    int val;
    int id;
}s[maxn];
int n,m;
int ans[maxn];
int h[maxn],cnt;
queue<X>Q[];

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) scanf("%d",&s[i].val);
    for(int i=;i<=n;i++) s[i].id=i;
    for(int i=;i<=n;i++)
    {
        int f; scanf("%d",&f); f++;
        Q[].push(s[f]);
    }

    int now=;

    while()
    {
        memset(h,,sizeof h);
        if(Q[now].size()==)
        {
            ans[Q[now].front().id]=;
            break;
        }

        while()
        {
            if(Q[now].size()<=m)
            {

                X tmp; tmp.val=-;
                while(!Q[now].empty())
                {
                    h[Q[now].front().id]=;
                    if(Q[now].front().val>tmp.val) tmp=Q[now].front();
                    Q[now].pop();
                }
                h[tmp.id]=;
                Q[now^].push(tmp);
                break;
            }
            else
            {
                X tmp; tmp.val=-;
                for(int i=;i<m;i++)
                {
                    h[Q[now].front().id]=;
                    if(Q[now].front().val>tmp.val) tmp=Q[now].front();
                    Q[now].pop();
                }
                h[tmp.id]=;
                Q[now^].push(tmp);
            }
        }
        now=now^;
        for(int i=;i<=n;i++)
            if(h[i]==) ans[i]=Q[now].size()+;
    }
    for(int i=;i<=n;i++)
    {
        printf("%d",ans[i]);
        if(i<n) printf(" ");
        else printf("\n");
    }
    return ;
}