#include <bits/stdc++.h> using namespace std; int main() { long long n,k,a,iloscroznych=0,lewo=0,wynik=0,ilosczmian=1,poz=0; int *tab = new int [500005]; int *ilosc = new int[500005]; int *lewa = new int [500005]; int *uzyte = new int [500005]; for(int i=0; i<500005; i++) { tab[i]=0; ilosc[i]=0; lewa[i]=0; uzyte[i]=0; } cin>>n>>k; for(int i=0; i<n; i++) { cin>>a; tab[i]=a; ilosc[a]++; } for(int i=0; i<500005; i++) { if(ilosc[i]>0) iloscroznych++; } if(iloscroznych>=k) { for(int i=0; i<n; i++) { if(lewa[tab[i]]==0) { lewa[tab[i]]++; lewo++; } else break; } poz=lewo; if(lewo>=k) cout<<0; else { for(int i=0; i<n; i++) { if(uzyte[tab[i]]==0 && ilosczmian<=k) { wynik+=(i+1-ilosczmian); uzyte[tab[i]]=1; ilosczmian++; } } cout<<wynik; } } 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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | #include <bits/stdc++.h> using namespace std; int main() { long long n,k,a,iloscroznych=0,lewo=0,wynik=0,ilosczmian=1,poz=0; int *tab = new int [500005]; int *ilosc = new int[500005]; int *lewa = new int [500005]; int *uzyte = new int [500005]; for(int i=0; i<500005; i++) { tab[i]=0; ilosc[i]=0; lewa[i]=0; uzyte[i]=0; } cin>>n>>k; for(int i=0; i<n; i++) { cin>>a; tab[i]=a; ilosc[a]++; } for(int i=0; i<500005; i++) { if(ilosc[i]>0) iloscroznych++; } if(iloscroznych>=k) { for(int i=0; i<n; i++) { if(lewa[tab[i]]==0) { lewa[tab[i]]++; lewo++; } else break; } poz=lewo; if(lewo>=k) cout<<0; else { for(int i=0; i<n; i++) { if(uzyte[tab[i]]==0 && ilosczmian<=k) { wynik+=(i+1-ilosczmian); uzyte[tab[i]]=1; ilosczmian++; } } cout<<wynik; } } else cout<<-1; return 0; } |