#include <bits/stdc++.h> using namespace std; void solve() { int n, k; cin >> n >> k; vector<int> v(n); for (int i = 0; i < n; i++) { cin >> v[i]; } vector<int> used(n+1); long long ans = 0; int curr = 0; for (int i = 0; i < n; i++) { if (!used[v[i]]) { used[v[i]]++; ans += i - curr; curr++; if (curr == k) { cout << ans << endl; return; } } } cout << -1 << endl; } int main() { ios_base::sync_with_stdio(false); cout.precision(12); cout << fixed; cin.tie(0); int T = 1; //cin >> T; for (int t = 0; t < T; t++) { solve(); } 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 | #include <bits/stdc++.h> using namespace std; void solve() { int n, k; cin >> n >> k; vector<int> v(n); for (int i = 0; i < n; i++) { cin >> v[i]; } vector<int> used(n+1); long long ans = 0; int curr = 0; for (int i = 0; i < n; i++) { if (!used[v[i]]) { used[v[i]]++; ans += i - curr; curr++; if (curr == k) { cout << ans << endl; return; } } } cout << -1 << endl; } int main() { ios_base::sync_with_stdio(false); cout.precision(12); cout << fixed; cin.tie(0); int T = 1; //cin >> T; for (int t = 0; t < T; t++) { solve(); } return 0; } |