#include <bits/stdc++.h>
#define rep(i, a, b) for (int i = a; i <= b; i++)
#define per(i, a, b) for (int i = b; a <= i; i--)
#define cat(x) cerr << #x << " = " << x << '\n';
using ll = long long;
using namespace std;
const int N = 500500;
int n, k, vis[N];
int main() {
cin.tie(0)->sync_with_stdio(0);
cin >> n >> k;
ll res = -1ll * k * (k + 1) / 2;
rep(i, 1, n) {
int x;
cin >> x;
if (!vis[x] && k) {
vis[x] = 1;
k--;
res += i;
}
}
if (k) {
res = -1;
}
cout << res << '\n';
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 | #include <bits/stdc++.h> #define rep(i, a, b) for (int i = a; i <= b; i++) #define per(i, a, b) for (int i = b; a <= i; i--) #define cat(x) cerr << #x << " = " << x << '\n'; using ll = long long; using namespace std; const int N = 500500; int n, k, vis[N]; int main() { cin.tie(0)->sync_with_stdio(0); cin >> n >> k; ll res = -1ll * k * (k + 1) / 2; rep(i, 1, n) { int x; cin >> x; if (!vis[x] && k) { vis[x] = 1; k--; res += i; } } if (k) { res = -1; } cout << res << '\n'; return 0; } |
English