//Ada Kołodziejczak //zadanie Oranżada #include<bits/stdc++.h> using namespace std; long long n, k, wyn, l, a; bitset<500005> bs; int main() { ios_base::sync_with_stdio(0); cin>>n>>k; for(int i=1; i<=n; ++i) { cin>>a; if(!bs[a]&&l<k) { wyn+=i-l-1; ++l; bs[a]=1; } } if(l!=k) cout<<-1; else cout<<wyn; }
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 | //Ada Kołodziejczak //zadanie Oranżada #include<bits/stdc++.h> using namespace std; long long n, k, wyn, l, a; bitset<500005> bs; int main() { ios_base::sync_with_stdio(0); cin>>n>>k; for(int i=1; i<=n; ++i) { cin>>a; if(!bs[a]&&l<k) { wyn+=i-l-1; ++l; bs[a]=1; } } if(l!=k) cout<<-1; else cout<<wyn; } |