1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include <bits/stdc++.h>

using namespace std;

bool eq(const vector<int>& tab, int l,int p){
    int x = tab[l];
    for(int i=l + 1;i<=p;i++)
        if(tab[i] != x)
            return false;
    return true;
}

int partition(vector<int>& tab, int l, int p, int k){
    if(eq(tab,l,p))
        return l;
    int x = rand()% (p - l + 1) + l;
    int y = p;
    swap(tab[x],tab[l]);
    x = l;
    while(l < p){
        if(tab[l] <= tab[x]){
            l++;
        }
        else{
            swap(tab[l],tab[p]);
            p--;
        }
    }
    if(tab[l] > tab[x]) l--;
    if((y - l - 1) >= k )
        return partition(tab,l+1,y,k);
    return partition(tab,x,l,k - (y - l));
}

int main(){
    vector<int> tab;
    int n,k,a;
    cin >> n >> k;
    for(int i=0; i < n; i ++){
        cin >> a;
        tab.push_back(a);
    }
    a = partition(tab,0,tab.size() - 1,k - 1);
    cout << n - a << endl;
}