#include <iostream> #include <vector> #include <algorithm> using namespace std; bool B[500004]; int T[500004]; int main() { std::ios::sync_with_stdio(false); cin.tie(NULL); int n,k; cin >> n >> k; for (int i=0;i<=n;i++) B[i] = false; for (int i=0;i<n;i++) cin >> T[i]; long long result = 0; int i; for (i=0;i<n && k>0;i++) { int a = T[i]; if (B[a]) break; k--; B[a]=true; } int differ = i+1; while (differ < n && k>0) { if (B[T[differ]]) {differ++; continue;} result += differ - i; B[T[differ]] = true; i++; differ++; k--; } if (k!=0) result = -1; cout << result << '\n'; }
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 | #include <iostream> #include <vector> #include <algorithm> using namespace std; bool B[500004]; int T[500004]; int main() { std::ios::sync_with_stdio(false); cin.tie(NULL); int n,k; cin >> n >> k; for (int i=0;i<=n;i++) B[i] = false; for (int i=0;i<n;i++) cin >> T[i]; long long result = 0; int i; for (i=0;i<n && k>0;i++) { int a = T[i]; if (B[a]) break; k--; B[a]=true; } int differ = i+1; while (differ < n && k>0) { if (B[T[differ]]) {differ++; continue;} result += differ - i; B[T[differ]] = true; i++; differ++; k--; } if (k!=0) result = -1; cout << result << '\n'; } |