#include <iostream> #include <vector> #define lli long long int using namespace std; int a, b, c, d, e, l, k, p, w, n, m, q; int czy[500001]; vector <int> poz; int main() { scanf("%d%d", &n, &k); for (a = 0; a != n; ++a) { scanf("%d", &l); if (!czy[l] && p < k) { w += a - p; ++p; czy[l] = 1; } } if (p < k) { printf("-1"); return 0; } printf("%d", w); }
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 | #include <iostream> #include <vector> #define lli long long int using namespace std; int a, b, c, d, e, l, k, p, w, n, m, q; int czy[500001]; vector <int> poz; int main() { scanf("%d%d", &n, &k); for (a = 0; a != n; ++a) { scanf("%d", &l); if (!czy[l] && p < k) { w += a - p; ++p; czy[l] = 1; } } if (p < k) { printf("-1"); return 0; } printf("%d", w); } |