#include <bits/stdc++.h>
using namespace std;
#define TESTS 0
const int N = 500100;
int n, k, a[N], cnt[N];
void solve() {
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> a[i];
}
vector<int> poses;
long long ans = 0;
for (int i = 1; i <= k; i++) {
if (cnt[a[i]]) {
poses.push_back(i);
}
else
cnt[a[i]] = 1;
}
reverse(poses.begin(), poses.end());
for (int i = k+1; i <= n; i++) {
if (poses.empty())
break;
if (cnt[a[i]] == 0) {
cnt[a[i]] = 1;
ans += i-poses.back();
poses.pop_back();
}
}
if (!poses.empty())
cout << -1 << "\n";
else
cout << ans << "\n";
}
int main() {
int tt = 1;
#if TESTS
cin >> tt;
#endif
while(tt--) {
solve();
}
}
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 | #include <bits/stdc++.h> using namespace std; #define TESTS 0 const int N = 500100; int n, k, a[N], cnt[N]; void solve() { cin >> n >> k; for (int i = 1; i <= n; i++) { cin >> a[i]; } vector<int> poses; long long ans = 0; for (int i = 1; i <= k; i++) { if (cnt[a[i]]) { poses.push_back(i); } else cnt[a[i]] = 1; } reverse(poses.begin(), poses.end()); for (int i = k+1; i <= n; i++) { if (poses.empty()) break; if (cnt[a[i]] == 0) { cnt[a[i]] = 1; ans += i-poses.back(); poses.pop_back(); } } if (!poses.empty()) cout << -1 << "\n"; else cout << ans << "\n"; } int main() { int tt = 1; #if TESTS cin >> tt; #endif while(tt--) { solve(); } } |
English