Passo 20

Instruções

Após o loop while, crie uma declaração if que verifique se current_node é None. Quando isso acontece, a lista ligada foi percorrida sem encontrar o elemento para remover e a função deve parar de executar. Portanto, adicione a palavra-chave return dentro do corpo do if.

O que fazer:

Testes:

Você deve criar uma declaração `if` que verifique se `current_node` é `None` e adicionar a palavra-chave `return` dentro dela.

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

        


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