#include <bits/stdc++.h> using namespace std; typedef long long LL; bool t[1000009]; bool d[1000009]; int main(){ ios_base::sync_with_stdio(false); int n, k, s=0; LL res = 0; cin>>n>>k; for(int i=0; i<n; i++){ int x; cin>>x; if(t[x]==false){ t[x]=true; d[i]=true; s++; } } for(int i=0, j=k; i<k; i++){ if(d[i]==false){ for(;j<n; j++){ if(d[j]==true) break; } res+=j-i; j++; } } if(s<k){ cout<<-1<<"\n"; }else{ cout<<res<<"\n"; } return 0; }
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 | #include <bits/stdc++.h> using namespace std; typedef long long LL; bool t[1000009]; bool d[1000009]; int main(){ ios_base::sync_with_stdio(false); int n, k, s=0; LL res = 0; cin>>n>>k; for(int i=0; i<n; i++){ int x; cin>>x; if(t[x]==false){ t[x]=true; d[i]=true; s++; } } for(int i=0, j=k; i<k; i++){ if(d[i]==false){ for(;j<n; j++){ if(d[j]==true) break; } res+=j-i; j++; } } if(s<k){ cout<<-1<<"\n"; }else{ cout<<res<<"\n"; } return 0; } |