#include<bits/stdc++.h>
using namespace std;
int main(){
int n,k,i,j;
long long result;
scanf("%d%d",&n,&k);
int arr[n];
bool bylo[n+1];
for(i=0;i<n;i++)
scanf("%d",&arr[i]);
for(i=1;i<=n;i++)
bylo[i]=false;
result=0;
j=0;
for(i=0;i<k;i++){
while(j<n && bylo[arr[j]])
j++;
if(j>=n)
break;
//printf(" %d %d\n",j,arr[j]);
result+=j-i;
bylo[arr[j]]=true;
}
if(j<n)
printf("%lld\n",result);
else
printf("-1\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 | #include<bits/stdc++.h> using namespace std; int main(){ int n,k,i,j; long long result; scanf("%d%d",&n,&k); int arr[n]; bool bylo[n+1]; for(i=0;i<n;i++) scanf("%d",&arr[i]); for(i=1;i<=n;i++) bylo[i]=false; result=0; j=0; for(i=0;i<k;i++){ while(j<n && bylo[arr[j]]) j++; if(j>=n) break; //printf(" %d %d\n",j,arr[j]); result+=j-i; bylo[arr[j]]=true; } if(j<n) printf("%lld\n",result); else printf("-1\n"); return 0; } |
English