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