#include<iostream> #include<vector> #include<algorithm> #include<unordered_set> using namespace std; int main(){ int n, k; cin >> n >> k; vector<int> bottles(n); for(int i=0;i<n;i++){ cin >> bottles[i]; } unordered_set<int> seen; long long count = 0; for(int i=0;i<n;i++){ if(seen.find(bottles[i]) == seen.end()){ seen.insert(bottles[i]); count += (i - seen.size() + 1); } if(seen.size() == k){ break; } } if(seen.size() < k){ cout << -1 << endl; return 0; } cout << count << endl; }
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 | #include<iostream> #include<vector> #include<algorithm> #include<unordered_set> using namespace std; int main(){ int n, k; cin >> n >> k; vector<int> bottles(n); for(int i=0;i<n;i++){ cin >> bottles[i]; } unordered_set<int> seen; long long count = 0; for(int i=0;i<n;i++){ if(seen.find(bottles[i]) == seen.end()){ seen.insert(bottles[i]); count += (i - seen.size() + 1); } if(seen.size() == k){ break; } } if(seen.size() < k){ cout << -1 << endl; return 0; } cout << count << endl; } |