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