#include <iostream> using namespace std; int n,k,j,i; int t[1000005]; bool zlicz[1000005]; long long sum; int main() { std::ios_base::sync_with_stdio(0); cin>>n>>k; for(i=1;i<=n;i++) { cin>>t[i]; } i=1; j=1; while(j<=k&&i<=n) { if(!zlicz[t[i]]) { sum+=(i-j); j++; zlicz[t[i]]=true; } i++; } if(j>k)cout<<sum; else cout<<"-1"; 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 | #include <iostream> using namespace std; int n,k,j,i; int t[1000005]; bool zlicz[1000005]; long long sum; int main() { std::ios_base::sync_with_stdio(0); cin>>n>>k; for(i=1;i<=n;i++) { cin>>t[i]; } i=1; j=1; while(j<=k&&i<=n) { if(!zlicz[t[i]]) { sum+=(i-j); j++; zlicz[t[i]]=true; } i++; } if(j>k)cout<<sum; else cout<<"-1"; return 0; } |