Source Code
#include "bits/stdc++.h"
using namespace std;
typedef long long ll;

int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    int n,k;
    cin>>n>>k;

    vector<int>freq(1e5+1),a(n);
    for(auto &x:a){
        cin>>x;
        freq[x]++;
    }

    vector<int>ans[k];
    for(int i=0;i<=1e5;i++){
        for(int j=0;j<freq[i];j++){
            ans[j%k].push_back(i);
        }
    }

    ll sum=0;
    for(int i=0;i<k;i++){
        sort(ans[i].begin(),ans[i].end());
        ans[i].resize(unique(ans[i].begin(),ans[i].end())-ans[i].begin());
        int mex=0;
        for(int j=0;j<ans[i].size();j++){
            if(j!=ans[i][j]) break;
            mex++;
        }
        sum+=mex;
    }

    cout<<sum;
}
Copy
Bitar The Handy Man YazanIstatiyeh
GNU G++17
9 ms
2.2 MB
Wrong Answer