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