//
#include <bits/stdc++.h>
using namespace std;
#define FOR(n, u) for(int (n) = 1; (n) <= (u); n++)
int main(){
// ios_base::sync_with_stdio(0);
// cin.tie(0);
// cout.tie(0);
int n, k, a;
cin >> n >> k;
list <int> lista;
bool czy[n + 1];
FOR(i, n){
czy[i] = 0;
}
int rozne = 0;
FOR(i, n){
cin >> a;
if(czy[a] == 0){
czy[a] = 1;
rozne++;
}
else{
lista.push_back(i);
}
}
if(rozne < k){
cout << -1;
return 0;
}
rozne = 0; int i = 1, il_do_przesuniecia = 0;
auto wsk = lista.begin();
while(rozne < k){
if(*wsk == i){
wsk++;
il_do_przesuniecia++;
rozne--;
}
rozne++;
i++;
}
long long suma = 0;
i--;
wsk--;
// cout << *wsk << endl;
while(il_do_przesuniecia > 0){
suma += i - *wsk;
//cout << i << " i " << * wsk << " -> " << suma << endl;
wsk--;
i--;
il_do_przesuniecia--;
}
cout << suma << endl;
return 0;
}
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | // #include <bits/stdc++.h> using namespace std; #define FOR(n, u) for(int (n) = 1; (n) <= (u); n++) int main(){ // ios_base::sync_with_stdio(0); // cin.tie(0); // cout.tie(0); int n, k, a; cin >> n >> k; list <int> lista; bool czy[n + 1]; FOR(i, n){ czy[i] = 0; } int rozne = 0; FOR(i, n){ cin >> a; if(czy[a] == 0){ czy[a] = 1; rozne++; } else{ lista.push_back(i); } } if(rozne < k){ cout << -1; return 0; } rozne = 0; int i = 1, il_do_przesuniecia = 0; auto wsk = lista.begin(); while(rozne < k){ if(*wsk == i){ wsk++; il_do_przesuniecia++; rozne--; } rozne++; i++; } long long suma = 0; i--; wsk--; // cout << *wsk << endl; while(il_do_przesuniecia > 0){ suma += i - *wsk; //cout << i << " i " << * wsk << " -> " << suma << endl; wsk--; i--; il_do_przesuniecia--; } cout << suma << endl; return 0; } |
English