#include <bits/stdc++.h> using namespace std; int a,n,k,ile,wyn,t[1048580]; bool u[2][500001]; void akt(int w)//drzewo punkt { while(w>1) { w=w>>1; t[w]=min(t[w<<1],t[(w<<1)+1]); } } int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); cin>>n>>k; n++; for(int i=1;i<n;++i) { cin>>a; if(u[0][a]!=true) { if(i!=1)t[i+524287]=i; else t[i+524287]=500005; u[0][a]=true; u[1][i]=true; ile++; } else{t[i+524287]=500005;} } for(int i=n+524287;i<1048580;++i){t[i]=500005;} for(int i=524287;i>0;--i){t[i]=min(t[i<<1],t[(i<<1)+1]);} if(ile<k){cout<<-1;} else { for(int K=1;K<n;++K) { if(u[1][K]!=true) { while(t[1]<500001&&K<=k) { //cout<<K<<" "<<t[1]<<"\n"; wyn+=(t[1]-K); K++; t[t[1]+524287]=500005; akt(t[1]+524287); } K=n+1; } } 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | #include <bits/stdc++.h> using namespace std; int a,n,k,ile,wyn,t[1048580]; bool u[2][500001]; void akt(int w)//drzewo punkt { while(w>1) { w=w>>1; t[w]=min(t[w<<1],t[(w<<1)+1]); } } int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); cin>>n>>k; n++; for(int i=1;i<n;++i) { cin>>a; if(u[0][a]!=true) { if(i!=1)t[i+524287]=i; else t[i+524287]=500005; u[0][a]=true; u[1][i]=true; ile++; } else{t[i+524287]=500005;} } for(int i=n+524287;i<1048580;++i){t[i]=500005;} for(int i=524287;i>0;--i){t[i]=min(t[i<<1],t[(i<<1)+1]);} if(ile<k){cout<<-1;} else { for(int K=1;K<n;++K) { if(u[1][K]!=true) { while(t[1]<500001&&K<=k) { //cout<<K<<" "<<t[1]<<"\n"; wyn+=(t[1]-K); K++; t[t[1]+524287]=500005; akt(t[1]+524287); } K=n+1; } } cout<<wyn; } } |