#include<iostream> using namespace std; int tab[500007]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n,k,m,l=0; long long sum=0,ile=0; cin>>n>>k; for(int i=1;i<=n;i++) { cin>>m; if(tab[m]==0 && ile<k) { ile++; tab[m]=1; l++; sum+=i-l; } } if(ile<k) { cout<<"-1"; return 0; }else{ cout<<sum; } }
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 | #include<iostream> using namespace std; int tab[500007]; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n,k,m,l=0; long long sum=0,ile=0; cin>>n>>k; for(int i=1;i<=n;i++) { cin>>m; if(tab[m]==0 && ile<k) { ile++; tab[m]=1; l++; sum+=i-l; } } if(ile<k) { cout<<"-1"; return 0; }else{ cout<<sum; } } |