P3290 寻找第K大数

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
using namespace std;
int a[40000],b[40000];
int n,k,maxn;
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1,x;i<=n;i++)
    {
        scanf("%d",&x);
        maxn=max(maxn,x);
        if(x>=0)    a[x]++;
        else b[-x]++;
    }
    for(int i=maxn;i>=0;i--)
    {
        k-=a[i];
        
        if(k<=0)    
        {
            printf("%d",i);
            return 0;
        }
    }
    for(int i=1;i<=40000;i++)
    {
        k-=b[i];        
        if(k<=0)    
        {
            printf("%d",-i);
            return 0;
        }
    }
    return 0;
}