#include <iostream>
#include <set>
#include <algorithm>
using namespace std;
int foo(int* start, int* end) {
int ret = 0;
for(int* i = start+1; i < end; ++i) {
if(*start != *i) {
swap(*start, *i);
return ret;
}
ret++;
}
return 0;
}
int main() {
ios::sync_with_stdio(0);
cout.tie();
cin.tie();
auto* s = new set<int>;
int n, k, answer = 0;
cin >> n >> k;
int arr[n];
for(int& i : arr) {
cin >> i;
s->insert(i);
}
if(s->size() < k) {
cout << "-1\n";
return 0;
}
delete s;
for(int* i = arr; i < arr+k; ++i) {
answer += foo(i, arr+n);
}
cout << answer << '\n';
}
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 | #include <iostream> #include <set> #include <algorithm> using namespace std; int foo(int* start, int* end) { int ret = 0; for(int* i = start+1; i < end; ++i) { if(*start != *i) { swap(*start, *i); return ret; } ret++; } return 0; } int main() { ios::sync_with_stdio(0); cout.tie(); cin.tie(); auto* s = new set<int>; int n, k, answer = 0; cin >> n >> k; int arr[n]; for(int& i : arr) { cin >> i; s->insert(i); } if(s->size() < k) { cout << "-1\n"; return 0; } delete s; for(int* i = arr; i < arr+k; ++i) { answer += foo(i, arr+n); } cout << answer << '\n'; } |
English