#include <iostream> using namespace std; int n, k; bool bylo[500005]; int tab[500005]; int main() { cin >> n >> k; for (int i = 0; i < n; i++) cin >> tab[i]; long long w = 0; bool czy = 0; int j = 1; for (int i = 0; i < k; i++, j++) { if (bylo[tab[i]] == true) { while (j <= n) { if (j < n && bylo[tab[j]] == false) { swap(tab[i], tab[j]); w += (j - i); break; } j++; } } if (j == n + 1) { czy = 1; break; } bylo[tab[i]] = true; } if (czy) cout << "-1"; else cout << w; 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 | #include <iostream> using namespace std; int n, k; bool bylo[500005]; int tab[500005]; int main() { cin >> n >> k; for (int i = 0; i < n; i++) cin >> tab[i]; long long w = 0; bool czy = 0; int j = 1; for (int i = 0; i < k; i++, j++) { if (bylo[tab[i]] == true) { while (j <= n) { if (j < n && bylo[tab[j]] == false) { swap(tab[i], tab[j]); w += (j - i); break; } j++; } } if (j == n + 1) { czy = 1; break; } bylo[tab[i]] = true; } if (czy) cout << "-1"; else cout << w; return 0; } |