PAT (Advanced Level) Practice 1129 Recommendation System （25 分）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Nightmare_ak/article/details/84873935

#include<cstdio>
#include<cmath>
#include<queue>
#include<set>
using namespace std;

const int N=5e4+5;

int time[N],res[N],in[N];

struct Node
{
    int id;
    bool operator<(const Node&p)const
    {
        if(time[id]!=time[p.id]) return time[id]>time[p.id];
        return id<p.id;
    }
};

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    set<Node> q;
    for(int i=1;i<=n;i++)
    {
        int x;scanf("%d",&x);
        if(i!=1)
        {
            int cnt=0;
            for(set<Node>::iterator it=q.begin();it!=q.end()&&cnt!=k;it++)
                res[++cnt]=it->id;
            printf("%d:",x);
            for(int i=1;i<=cnt;i++)
                printf(" %d",res[i]);
            puts("");
        }
        if(!in[x]) q.insert({x}),in[x]=1;
        else
        {
            q.erase(q.find({x}));
            time[x]++;
            q.insert({x});
        }
    }
    return 0;
}