#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; } |
English