Bear and Displayed Friends

题目描述

输入格式

输出格式

 样例1

Input

Output

样例2 

Input

 Output

Note 

 代码

#include <iostream>
#include <algorithm>
using namespace std;
int ti[150010];
bool st[150010];
typedef pair<int,int>PII;
PII f[7];
 
void qsort(PII *a,int n)
{
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            if(f[i].first>f[j].first)
            {
                swap(f[i].first,f[j].first);
                swap(f[i].second,f[j].second);
            }
        }
    }
}
 
int main()
{
    int n,k,q;
    cin>>n>>k>>q;
    for(int i=1;i<=n;i++)cin>>ti[i];
    int op,number,s=0;
    while(q--)
    {
        cin>>op>>number;
        if(op==1)
        {
            st[number]=1;
            s++;
            if(s>k)
            {
                qsort(f,k);
                for(int i=1;i<=k;i++)
                {
                    if(ti[number]>f[i].first)
                    {
                        f[i].first=ti[number];
                        f[i].second=number;
                        break;
                    }
                }
            }
            else
            {
                f[s].first=ti[number];
                f[s].second=number;
            }
        }
        if(op==2)
        {
            if(st[number])
            {
                bool flag=0;
                for(int i=1;i<=k;i++)
                {
                    if(f[i].second==number)
                    {
                        flag=1;
                        break;
                    }
                }
                if(flag)cout<<"YES"<<endl;
                else cout<<"NO"<<endl;
            }
            else cout<<"NO"<<endl;
        }
    }
    return 0;
}