#include <iostream> #include <vector> #include <stdint.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); uint32_t n,k; cin>>n>>k; vector<int> marki(n,0); for (uint32_t ii=0; ii<n; ii++) { cin>>marki[ii]; } vector<bool> czyDodana(n+1, false); uint64_t licznik = 0; uint32_t jj=0; for (uint32_t ii=0; ii<k; ii++) { if (!czyDodana[marki[ii]]) czyDodana[marki[ii]] = true; else // powtorzona marka wsrod pierwszych k butelek { if (jj<ii) jj=ii+1; while (jj<n && czyDodana[marki[jj]]) jj++; if (jj==n) // wyjscie poza tablice, nie ma z czym zamienic powtorzonej butelki { cout<<"-1"<<endl; return 0; } else { czyDodana[marki[jj]] = true; licznik += (jj-ii); } } } cout<<licznik<<endl; 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 | #include <iostream> #include <vector> #include <stdint.h> using namespace std; int main() { ios_base::sync_with_stdio(0); cin.tie(0); uint32_t n,k; cin>>n>>k; vector<int> marki(n,0); for (uint32_t ii=0; ii<n; ii++) { cin>>marki[ii]; } vector<bool> czyDodana(n+1, false); uint64_t licznik = 0; uint32_t jj=0; for (uint32_t ii=0; ii<k; ii++) { if (!czyDodana[marki[ii]]) czyDodana[marki[ii]] = true; else // powtorzona marka wsrod pierwszych k butelek { if (jj<ii) jj=ii+1; while (jj<n && czyDodana[marki[jj]]) jj++; if (jj==n) // wyjscie poza tablice, nie ma z czym zamienic powtorzonej butelki { cout<<"-1"<<endl; return 0; } else { czyDodana[marki[jj]] = true; licznik += (jj-ii); } } } cout<<licznik<<endl; return 0; } |