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