Passo 24

Instruções

Agora, é hora de testar a remoção de um elemento da sua lista ligada. No final do seu código, remova 1 da lista usando o método remove. Em seguida, imprima my_list.length novamente para ver o comprimento atualizado da lista ligada. Com isso, a oficina sobre a estrutura de dados lista ligada está completa!

O que fazer:

Testes:

Você deve remover `1` da lista usando o método `remove`.
Você deve imprimir `my_list.length`.

class LinkedList:
    class Node:
        def __init__(self, element):
            self.element = element
            self.next = None
            
    def __init__(self):
        self.length = 0
        self.head = None

    def is_empty(self):
        return self.length == 0
    
    def add(self, element):
        node = self.Node(element)
        if self.is_empty():
            self.head = node
        else:
            current_node = self.head
            while current_node.next is not None:
                current_node = current_node.next
            current_node.next = node
        self.length += 1

    def remove(self, element):
        previous_node = None
        current_node = self.head
        while current_node is not None and current_node.element != element:
            previous_node = current_node
            current_node = current_node.next
        if current_node is None:
            return        
        elif previous_node is not None:
            previous_node.next = current_node.next
        else:
            self.head = current_node.next
        self.length -= 1

my_list = LinkedList()
print(my_list.is_empty())

my_list.add(1)
my_list.add(2)
print(my_list.is_empty())
print(my_list.length)

Preview