Codeforces 639A Bear and Displayed Friends （优先队列）

题目链接：点击打开链接

#include<stdio.h>
#include<queue>
using namespace std;
struct node{
int id,val;
}num[150010];
int visit[150010];
bool operator<(const node &a,const node &b)
{
return a.val>b.val;
}
int main()
{
int n,k,m,i,op,x;
node now;
while(~scanf("%d%d%d",&n,&k,&m))
{
for(i=1;i<=n;i++)
{
      scanf("%d",&num[i].val);
      num[i].id=i;
      visit[i]=0;
}
priority_queue<node>q;
while(m--)
{
scanf("%d%d",&op,&x);
if(op==1)
{
q.push(num[x]);
visit[x]=1;
if(q.size()>k)
{
visit[q.top().id]=0;
q.pop();
}

}
else
if(visit[x]==1)
printf("YES\n");
else
printf("NO\n");
}
}
return 0;
}