Data Science Academy - Python Fundamentos - Capítulo 7

Download: http://github.com/dsacademybr


In [1]:
# Versão da Linguagem Python
from platform import python_version
print('Versão da Linguagem Python Usada Neste Jupyter Notebook:', python_version())


Versão da Linguagem Python Usada Neste Jupyter Notebook: 3.7.6

Missão 2: Implementar o Algoritmo de Ordenação "Selection sort".

Nível de Dificuldade: Alto

Premissas

  • As duplicatas são permitidas?      * Sim
  • Podemos assumir que a entrada é válida?      * Não
  • Podemos supor que isso se encaixa na memória?      * Sim

Teste Cases

  • None -> Exception
  • [] -> []
  • One element -> [element]
  • Two or more elements

Algoritmo

Animação do Wikipedia:

Podemos fazer isso de forma recursiva ou iterativa. Iterativamente será mais eficiente, pois não requer sobrecarga de espaço extra com as chamadas recursivas.

  • Para cada elemento      Verifique cada elemento à direita para encontrar o min      Se min < elemento atual, swap

Solução


In [2]:
class SelectionSort(object):

    def sort(self, data):
        if data is None:
            raise TypeError('Dados não podem ser None')
        if len(data) < 2:
            return data
        for i in range(len(data) - 1):
            min_index = i
            for j in range(i + 1, len(data)):
                if data[j] < data[min_index]:
                    min_index = j
            if data[min_index] < data[i]:
                data[i], data[min_index] = data[min_index], data[i]
        return data

    def sort_iterative_alt(self, data):
        if data is None:
            raise TypeError('Dados não podem ser None')
        if len(data) < 2:
            return data
        for i in range(len(data) - 1):
            self._swap(data, i, self._find_min_index(data, i))
        return data

    def sort_recursive(self, data):
        if data is None:
            raise TypeError('Dados não podem ser None')
        if len(data) < 2:
            return data
        return self._sort_recursive(data, start=0)

    def _sort_recursive(self, data, start):
        if data is None:
            return
        if start < len(data) - 1:
            swap(data, start, self._find_min_index(data, start))
            self._sort_recursive(data, start + 1)
        return data

    def _find_min_index(self, data, start):
        min_index = start
        for i in range(start + 1, len(data)):
            if data[i] < data[min_index]:
                min_index = i
        return min_index

    def _swap(self, data, i, j):
        if i != j:
            data[i], data[j] = data[j], data[i]
        return data

Teste da Solução


In [3]:
%%writefile missao4.py
from nose.tools import assert_equal, assert_raises


class TestSelectionSort(object):

    def test_selection_sort(self, func):
        print('None input')
        assert_raises(TypeError, func, None)

        print('Input vazio')
        assert_equal(func([]), [])

        print('Um elemento')
        assert_equal(func([5]), [5])

        print('Dois ou mais elementos')
        data = [5, 1, 7, 2, 6, -3, 5, 7, -10]
        assert_equal(func(data), sorted(data))

        print('Sua solução foi executada com sucesso! Parabéns!')


def main():
    test = TestSelectionSort()
    try:
        selection_sort = SelectionSort()
        test.test_selection_sort(selection_sort.sort)
    except NameError:
        pass


if __name__ == '__main__':
    main()


Overwriting missao4.py

In [4]:
%run -i missao4.py


None input
Input vazio
Um elemento
Dois ou mais elementos
Sua solução foi executada com sucesso! Parabéns!

Fim

Obrigado - Data Science Academy - facebook.com/dsacademybr