#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; } |