Koszulki

[PA2021] Koszulki

题目描述

$n$ 个人参加一场比赛，其中第 $i$ 个人的得分为 $a_i$。

主办方决定至少颁发 $k$ 件礼物。

但是，若 $\exists 1\le x,y\le n,a_x\ge a_y$ 且 $x$ 未拿到礼物但 $y$ 拿到礼物，$x$ 会不满意。

主办方希望每个人都满意，求颁发的礼物数量的最小值。

输入格式

第一行，两个整数 $n,k$。

第二行，$n$ 个整数 $a_1,a_2,\cdots ,a_n$。

输出格式

一行，一个整数，表示所求的值。

样例 #1

样例输入 #1

5 3
75 90 120 75 40

样例输出 #1

4

提示说明

样例 #1 解释

最优方案为除了最后一人以外全部送礼物。

数据范围

对于 $100\%$ 的数据，$1\le k\le n\le 2\times 10^3$，$1\le a_i\le 120$。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
ll a[150];

int main()
{
    ll n,k;
    cin>>n>>k;
    
    for(ll i=1;i<=n;i++)
    {
        ll x;
        cin>>x;
        a[x]++;
    }

    ll ans=0;
    for(ll i=120;i>0;i--)
    {
        ans+=a[i];
        if(ans>=k) break;
    }
    
    cout<<ans<<endl;
    return 0;
}