// #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; } |