#include <stdio.h>
#include <iostream>
#include <utility>
using namespace std;
int main() {
int n, k;
scanf("%d %d", &n, &k);
int data[n+1];
int pos[k];
int dif[n+1];
for (int i = 0; i < n; i++) {
dif[i] = 0;
}
for (int i = 0; i < n; i++) {
scanf("%d", &data[i]);
}
int counter = 0;
int it = 0;
while(counter < k && it < n) {
if (!dif[data[it]]++) {
pos[counter++] = it;
}
it++;
}
if (counter < k) {
printf("-1");
return 0;
}
long long sum = 0;
for (int i = 0; i < k; i++) {
sum += pos[i] - i;
}
printf("%lld", sum);
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 | #include <stdio.h> #include <iostream> #include <utility> using namespace std; int main() { int n, k; scanf("%d %d", &n, &k); int data[n+1]; int pos[k]; int dif[n+1]; for (int i = 0; i < n; i++) { dif[i] = 0; } for (int i = 0; i < n; i++) { scanf("%d", &data[i]); } int counter = 0; int it = 0; while(counter < k && it < n) { if (!dif[data[it]]++) { pos[counter++] = it; } it++; } if (counter < k) { printf("-1"); return 0; } long long sum = 0; for (int i = 0; i < k; i++) { sum += pos[i] - i; } printf("%lld", sum); return 0; } |
English