#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; } |