Strona główna » Algorytmy » Sortowanie » Sortowanie Tablicy Zer i Jedynek
 

Sortowanie Tablicy Zer i Jedynek

Zadanie

Zadanie polega na posortowaniu tablicy, której elementami są jedynie wartości 0 oraz 1. Przykładowo dla tablicy [1, 0, 1, 0, 1] poprawnym rozwiązaniem jest [0, 0, 1, 1, 1]. Zakładamy, że tablica ma co najmniej jeden element, ale jeden z elementów nie musi występować.

Analiza zadania

Zadanie to można rozwiązać używając dowolnego algorytmu do sortowania. Najrozsądniejszą opcją wydaje się tutaj sortowanie przez zliczanie, ponieważ z góry wiadomo, że są tylko dwa elementy do zliczania. Warto jednak zauważyć, że znając ilość np. cyfry zero wiadomo ile jest cyfr 1, więc można zrezygnować z dwóch liczników na rzecz jednego. W ten sposób tablica zostaje posortowane w czasie liniowym: pierwsze przejście zlicza ilość zer, a drugie przejście wpisuje do tablicy odpowiednią ilość zer oraz jedynek.

Jednak czy jest to na pewno najlepsze możliwe rozwiązanie? Każdy element jest co prawda odczytywany raz, ale być może tablica jest już posortowana i wtedy każdy element zostanie nadpisany choć nie musi.

Implementacja

Rozwiązanie proste

Poniższa funkcja sortuj() przyjmuje jako argument tablicę do posortowania tab składającą sie z samych 0 i 1.

C++C#
Python
  1. def sortuj(tab):
  2.   zer = 0
  3.   for i in range(len(tab)):
  4.     if (tab[i] == 0):
  5.       zer += 1
  6.   for i in range(0, zer):
  7.     tab[i] = 0
  8.   for i in range(zer, len(tab)):
  9.     tab[i] = 1

(2.) Na początku algorytmu deklarujemy licznik zer, który posłuży do zliczania zer w tablicy. Następnie (3. - 5.) algorytm sprawdza kolejny element. Po zliczeniu wszystkich zer tablica zostaje uzupełniona odpowiednią ilością (6. - 7.) zer i (8. - 9.) jedynek.

Rozwiązanie optymalne

Zauważmy, że do posortowania tablicy [1, 0, 1, 0, 1] wystarczy zamienić jedynie pierwszą wartość 1 z ostatnim 0. Podobnie jest w przypadku [1, 1, 0, 1, 0], gdzie taką zamianę trzeba wykonać dwa razy. Algorytm nie potrafi jak człowiek spojrzeć na tablicę i od razu znaleźć parę do zamiany, więc na przejrzeniu każdego kolejnego elementu nie uda się zaoszczędzić, ale zauważmy, że elementy stojące na swojej pozycji nie są nadpisywane, więc nie są wykonywane zbędne operacje zapisu.

Implmentacja takiego rozwiązania wymaga zadeklarowania dwóch wskaźników: lewego oraz prawego. Początkowy lewy wskaźnik wskazuje pierwszy element, a prawy ostatni. Lewy wskaźnik jest tak długo zwiększany, aż napotka wartość 1, a prawy tak długo zmniejszany, aż napotka wartość 0. W momencie znalezienia takiej pary elementów należy zamienić je miejscami i dokonać ponownego przeszukania tablicy. Warunkiem stopu jest spotkanie się liczników, że lewy będzie większy bądź równy prawemu.

C++C#
Python
  1. def sortuj(tab):
  2.   lewy, prawy = 0, len(tab) - 1
  3.   while (lewy < prawy):
  4.     while (tab[lewy] == 0 and lewy < prawy):
  5.       lewy += 1
  6.     while (tab[prawy] == 1 and lewy < prawy):
  7.       prawy -= 1
  8.     if (lewy < prawy):
  9.       tab[lewy] = 0
  10.       tab[prawy] = 1
  11.       lewy += 1
  12.       prawy -= 1

(2.) Na początku algorytmu należy ustalić wartości początkowe lewego oraz prawego brzegu. Następnie (3.) w pętli, aż nie zostanie spełniony warunek ostateczny to: (4. - 5.) wyszukujemy jedynki z lewej strony oraz (6. - 7.) 0 od prawej strony. (8.) Jeśli pomimo przesunięć wskaźników dalej spełniają warunek to (9. - 10.) elementy należy zamienić miejscami oraz (11. - 12.) przesunąć wskaźniki.

Testowanie funkcji

W celu przetestowania napisanej funkcji można skorzystać z poniższego fragmentu kodu:

C++C#
Python
  1. print("Podaj elementy tablicy:")
  2. tab = [int(x) for x in input().split(' ')]
  3. sortuj(tab)
  4. for i in range(0, len(tab)):
  5.   print(tab[i], end=' ')