#include "bits/stdc++.h" // Ignacy Boehlke using namespace std; // XIII LO Szczecin int main() { int n, k; scanf("%d%d", &n, &k); int cnt = 0; long long res = 0; vector<bool> vis(n); for (int i = 0; i < n; ++i) { int x; scanf("%d", &x); --x; if (!vis[x] && cnt < k) { vis[x] = true; res += i - cnt++; } } printf("%lld\n", (cnt < k ? -1LL : res)); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | #include "bits/stdc++.h" // Ignacy Boehlke using namespace std; // XIII LO Szczecin int main() { int n, k; scanf("%d%d", &n, &k); int cnt = 0; long long res = 0; vector<bool> vis(n); for (int i = 0; i < n; ++i) { int x; scanf("%d", &x); --x; if (!vis[x] && cnt < k) { vis[x] = true; res += i - cnt++; } } printf("%lld\n", (cnt < k ? -1LL : res)); } |