#include<bits/stdc++.h> using namespace std; using lld = long long; int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int n,k; cin >> n >> k; vector<int> t(n); for(int& i : t) cin >> i, i--; vector<bool> was(n,false); lld res = 0; int j = 0; for(int i=0;i<n && j<k;i++) if(!was[t[i]]) res += i-j, j++, was[t[i]] = true; if(j < k) cout << "-1\n"; else cout << res << "\n"; 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 | #include<bits/stdc++.h> using namespace std; using lld = long long; int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int n,k; cin >> n >> k; vector<int> t(n); for(int& i : t) cin >> i, i--; vector<bool> was(n,false); lld res = 0; int j = 0; for(int i=0;i<n && j<k;i++) if(!was[t[i]]) res += i-j, j++, was[t[i]] = true; if(j < k) cout << "-1\n"; else cout << res << "\n"; return 0; } |