#include <cstdio>
#include <vector>
int main() {
int N, K;
int result = 0;
int found = 0;
std::vector<bool> visited;
scanf("%d %d", &N, &K);
visited.resize(N+1, false);
for (int i=0; i<N; ++i) {
int x;
scanf("%d", &x);
if (visited[x] == false) {
result += (i-found);
found++;
}
if (found==K) {
break;
}
visited[x] = true;
}
if (K==found) {
printf("%d\n", result);
} else {
printf("-1\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 35 36 | #include <cstdio> #include <vector> int main() { int N, K; int result = 0; int found = 0; std::vector<bool> visited; scanf("%d %d", &N, &K); visited.resize(N+1, false); for (int i=0; i<N; ++i) { int x; scanf("%d", &x); if (visited[x] == false) { result += (i-found); found++; } if (found==K) { break; } visited[x] = true; } if (K==found) { printf("%d\n", result); } else { printf("-1\n"); } return 0; } |
English