#include <bits/stdc++.h> using namespace std; set <long long> butelki; long long n, k, a, wyn; int main() { ios_base::sync_with_stdio(false); cin.tie(0); cin >> n >> k; for(long long i=0; i<n && butelki.size()<k; i++) { cin >> a; if(!butelki.count(a)) { butelki.insert(a); wyn=wyn+i-butelki.size()+1; } } if(butelki.size()==k) cout<<wyn<<endl; else cout<<"-1"<<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 | #include <bits/stdc++.h> using namespace std; set <long long> butelki; long long n, k, a, wyn; int main() { ios_base::sync_with_stdio(false); cin.tie(0); cin >> n >> k; for(long long i=0; i<n && butelki.size()<k; i++) { cin >> a; if(!butelki.count(a)) { butelki.insert(a); wyn=wyn+i-butelki.size()+1; } } if(butelki.size()==k) cout<<wyn<<endl; else cout<<"-1"<<endl; } |