VK Cup 2016 - Round 1 (Div. 2 Edition) B B. Bear and Displayed Friends 优先队列

题目:给一个人对朋友打分的数组。一开始没人
然后又q个操作，
1. x ID为x的上线了
2. x 问ID为X是否在屏幕上
屏幕上只能有K个人，而且是上线的分数前K大的人
思路:维持一个数量<=k的优先队列就行了

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
#include<string>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) (x&(-x))
typedef long long LL;
#define maxn 150005
const int inf=(1<<28)-1;
int A[maxn];
struct node
{
    int val,pos;
    friend bool operator <(node u,node v)
    {
        return u.val>v.val;
    }
    node(int a=0,int b=0)
    {
        val=a;pos=b;
    }
};
priority_queue<node>Q;
bool mark[maxn];
int main()
{
    int n,q,k;
    scanf("%d%d%d",&n,&k,&q);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&A[i]);
    }
    for(int i=1;i<=q;++i)
    {
        int ope,Id;
        scanf("%d%d",&ope,&Id);
        if(ope==1)
        {
            Q.push(node(A[Id],Id));
            mark[Id]=1;
            if(Q.size()>k)
            {
                node cur=Q.top();
                mark[cur.pos]=0;
                Q.pop();
            }
        }
        else
        {
            if(mark[Id]) printf("YES\n");
            else printf("NO\n");
        }
    }
    return 0;
}