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

humeay

于 2016-03-30 21:37:31 发布

阅读量308

点赞数

CC 4.0 BY-SA版权

分类专栏： codeforces 其他文章标签：优先队列 codeforces

本文链接：https://blog.youkuaiyun.com/qq_27925701/article/details/51019631

codeforces 同时被 2 个专栏收录

32 篇文章

订阅专栏

其他

13 篇文章

订阅专栏

本文介绍了一个使用优先队列实现的在线算法，该算法能够实时维护一个最多包含K个元素的列表，这些元素是从一系列给定的分数中选择出来的，并且总是保持列表中的元素拥有最高的分数。此外，该算法还提供了一种查询指定元素是否存在于当前列表中的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目:给一个人对朋友打分的数组。一开始没人
然后又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;
}