VK Cup 2016 - Round 1 (Div. 2 Edition)-B - Bear and Displayed Friends-STL(维护前k大)


http://codeforces.com/contest/658/problem/B


题意 给n,k,q;


就是让你维护一个kmax的集合

q次操作

操作1 把 节点v加入kmax集合,

操作2 查询v是否在集合里

直接用multiset。就好了。。


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

const double pi=acos(-1.0);
double eps=0.000001; 
__int64 min(__int64 a,__int64 b)
{return a<b?a:b;} 
__int64 max(__int64 a,__int64 b)
{return a>b?a:b;}

struct node
{
	int x,val;
	node(){}
	node(int a,int b){x=a,val=b;}
	bool operator <(const node &b)  const
    {    
        return val<b.val;//最小值优先     
    }   
	bool operator ==(const node &b)  const
    {    
        return x==b.x;      
    }   
}; 
multiset<node> q;
 int n,k,qq;
 int t[150005];
 
			node tmp[8];
int main() 
{ int i,j;
 
	scanf("%d%d%d",&n,&k,&qq);
	for (i=1;i<=n;i++)
		scanf("%d",&t[i]);

	int cur=0,x,y;
	for (i=1;i<=qq;i++)
	{
		scanf("%d%d",&x,&y);
		if (x==1)
		{
			if (cur<k)
			{
				q.insert(node(y,t[y])); 
				cur++;
			}
			else
			{
				q.insert(node(y,t[y]));
			//	printf("%d\n",q.begin()->x);
				q.erase(q.begin());
			}
		}
		else
		{
			int flag=0;
			if (q.find(node(y,t[y]))!=q.end())
				flag=1;

			if (flag) printf("YES\n");
			else
				printf("NO\n"); 
		}

	}

	
	
	
	
	
	
	return 0;
	
}


你可能感兴趣的:(VK Cup 2016 - Round 1 (Div. 2 Edition)-B - Bear and Displayed Friends-STL(维护前k大))