#include <cstdio> using namespace std; #define MAX_N 500010 int N, K; int A[MAX_N], ai, lb; long long t; int main() { scanf("%d %d", &N, &K); for (int i = 0; K && i < N; ++i) { scanf("%d", &ai); if (!A[ai]) { A[ai] = 1; t += 1LL * (i - lb); lb++; K--; } } printf("%lld\n", K ? -1LL : t); 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 | #include <cstdio> using namespace std; #define MAX_N 500010 int N, K; int A[MAX_N], ai, lb; long long t; int main() { scanf("%d %d", &N, &K); for (int i = 0; K && i < N; ++i) { scanf("%d", &ai); if (!A[ai]) { A[ai] = 1; t += 1LL * (i - lb); lb++; K--; } } printf("%lld\n", K ? -1LL : t); return 0; } |