#include <iostream> constexpr int MAX_BUTELEK = 500000; int pierwsza[MAX_BUTELEK]; int liczba_roznych = 0; int indeksy_pierwszych[MAX_BUTELEK]; int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); int liczba_butelek, liczba_lewych; std::cin >> liczba_butelek >> liczba_lewych; for(int i = 0, marka; i < liczba_butelek; ++i) { std::cin >> marka; if(!pierwsza[marka-1]) { indeksy_pierwszych[liczba_roznych++] = i+1; pierwsza[marka-1] = i+1; } } if(liczba_roznych < liczba_lewych) std::cout << "-1\n"; else { long long wynik = 0; for(int i = 0; i < liczba_lewych; ++i) wynik += indeksy_pierwszych[i]-i-1; std::cout << wynik << '\n'; } 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 | #include <iostream> constexpr int MAX_BUTELEK = 500000; int pierwsza[MAX_BUTELEK]; int liczba_roznych = 0; int indeksy_pierwszych[MAX_BUTELEK]; int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); int liczba_butelek, liczba_lewych; std::cin >> liczba_butelek >> liczba_lewych; for(int i = 0, marka; i < liczba_butelek; ++i) { std::cin >> marka; if(!pierwsza[marka-1]) { indeksy_pierwszych[liczba_roznych++] = i+1; pierwsza[marka-1] = i+1; } } if(liczba_roznych < liczba_lewych) std::cout << "-1\n"; else { long long wynik = 0; for(int i = 0; i < liczba_lewych; ++i) wynik += indeksy_pierwszych[i]-i-1; std::cout << wynik << '\n'; } return 0; } |