#include<iostream>
#include<cstdlib>
#include<cstdio>
using namespace std;
int n,k,t[500005],q;
long long q2;
bool k2[500005];
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d",&t[i]);
if(k2[t[i]]==0){
q++;
if(i>k) q2+=i;
if(q==k) break;
}else{
if(i<=k) q2-=i;
}
k2[t[i]]=1;
}
if(q<k){
printf("-1");
return 0;
}
printf("%lld",q2);
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 | #include<iostream> #include<cstdlib> #include<cstdio> using namespace std; int n,k,t[500005],q; long long q2; bool k2[500005]; int main(){ scanf("%d%d",&n,&k); for(int i=1;i<=n;i++){ scanf("%d",&t[i]); if(k2[t[i]]==0){ q++; if(i>k) q2+=i; if(q==k) break; }else{ if(i<=k) q2-=i; } k2[t[i]]=1; } if(q<k){ printf("-1"); return 0; } printf("%lld",q2); return 0; } |
English