#include <iostream> using namespace std; int butelki[500002]; int main() { int n,k,wynik=0, poz=1; cin>>n>>k; for(int i=1; i<=n; i++) { cin>>butelki[i]; } int i=1; while(k>0 && i<=n) { if(butelki[i]!=0) { k--; wynik+=(i-poz); poz++; for(int j=i+1; j<=n; j++) { if(butelki[j]==butelki[i]) butelki[j]=0; } butelki[i]=0; } i++; } if(k==0) cout<<wynik; else cout<<-1; }
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 | #include <iostream> using namespace std; int butelki[500002]; int main() { int n,k,wynik=0, poz=1; cin>>n>>k; for(int i=1; i<=n; i++) { cin>>butelki[i]; } int i=1; while(k>0 && i<=n) { if(butelki[i]!=0) { k--; wynik+=(i-poz); poz++; for(int j=i+1; j<=n; j++) { if(butelki[j]==butelki[i]) butelki[j]=0; } butelki[i]=0; } i++; } if(k==0) cout<<wynik; else cout<<-1; } |