题目:
利马克是一只小北极熊。他喜欢通过社交网络与其他熊联系。他有n个朋友,他与其中第i个朋友的关系由一个独特的整数ti来描述。这个值越大,友谊就越好。没有两个朋友有相同的值ti。
春天开始了,熊的冬季睡眠结束了。利马克刚刚醒来并登录了。他所有的朋友都还在睡觉,因此他们都不在线。在接下来的几个小时里,他们中的一些(也许是全部)会出现在网上,一次一个。
系统显示正在在线的朋友。在屏幕上,最多可以显示k个朋友。如果有超过k个朋友在线,那么系统只显示其中最好的k个朋友--那些ti最大的朋友。
你的任务是处理两种类型的查询。
"1 id"--朋友id成为在线。保证他以前不在线。
"2 id"- 检查系统是否显示了朋友的ID。在另一行打印 "YES "或 "NO"。
你能帮助Limak并回答所有第二类的询问吗?
输入
第一行包含三个整数n、k和q(1≤n,q≤150000,1≤k≤min(6,n))--分别是朋友的数量、显示的最大在线朋友数量和查询的数量。
第二行包含n个整数t1,t2,...,tn(1≤ti≤1e9),其中ti描述Limak与第i个朋友的关系有多好。
下面q行中的第i行包含两个整数typei和idi(1≤typei≤2,1≤idi≤n)--第i个查询。如果typei=1,那么朋友idi成为在线。如果typei=2,那么你应该检查一个朋友idi是否被显示。
保证没有两个第一类型的查询会有相同的白痴,因为一个朋友不可能在线两次。另外,保证至少有一个查询是第二种类型的(typei=2),所以输出不会是空的。
输出
对于每个第二类型的查询,如果给定的朋友被显示,则打印一行答案--"YES"(不带引号),否则打印 "NO"(不带引号)。
思路详解:题目要查询指定朋友的状态,可以通过结构体把每个朋友的两个参量绑定来优化过程,因为题目中的n,k,q都在150000之内,时间复杂度不是很大,所以只要把状态处理好即可
本题考点:结构体
代码实现:
#include <bits/stdc++.h>
using namespace std;
struct pyq{
int yq;//友情
int zt;//状态
}py[150000];//定义一个结构体变量用来储存每个朋友的状态(zt)和友情(yq)大小
int main (){
int n,k,q,j=1;
cin>>n>>k>>q;
for(int i=1;i<=n;i++){//循环从1开始保证与朋友编号一致
cin>>py[i].yq;
py[i].zt=0; //将所有朋友的初始状态设置为0(即不在线)
}
int z,x,sum=0,min1=0x3f3f3f,min2=-1;
/*
此处sum用于判断上线人数是否超过k
min1用于记录最小友情值
min2记录最小友情值对应的朋友
*/
while(j<=q){
cin>>z>>x;
if(z==1){
py[x].zt=1;//朋友上线记状态为1
if(py[x].yq<min1){//刷新最小值
min1=py[x].yq;
min2=x;
}
sum++;//刷新上线人数
if(sum>k){
sum--;
py[min2].zt=2;//若上线人数超过k则将友情值最低的朋友的状态记为2
min1=0x3f3f3f;//将min1、min2回归初始状态防止干扰下次循环
min2=-1;
}
}
if(z==2){//指令为2时直接查找即可,对状态的操作都已在指令1中处理完毕
if(py[x].zt==1) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
j++;
}
return 0;
}
测试数据:
输入: 4 2 8 300 950 500 200 1 3 2 4 2 3 1 1 1 2 2 1 2 2 2 3输出:
NO YES NO YES YES