Question

我的家庭作业需要我从链接列表中弹出最后一个项目，由于某种原因，一些测试用例可以工作，但有些不能用，我也不知道为什么。

class Node:
    def __init__(self, init_data):
        self.data = init_data
        self.next = None
    def get_data(self):
        return self.data
    def get_next(self):
        return self.next
    def set_data(self, new_data):
        self.data = new_data
    def set_next(self, new_next):
        self.next = new_next
    def __str__(self):
        return str(self.data)

class LinkedList:
    def __init__(self):
        self.head = None
    def add(self, item):
        new_node = Node(item)
        new_node.set_next(self.head)
        self.head = new_node

    def __str__(self):
        result = "("
        node = self.head
        if node != None:
            result += str(node.data)
            node = node.next
            while node:
                result += ", " + str(node.data)
                node = node.next
        result += ")"
        return result
    def remove_from_tail(self):
        if self.head is None:
            return None
        prev = None
        cur = self.head
        while cur.next is not None:
            prev = cur
            cur = cur.next
        if prev:
            prev.next = None
        return cur

#test case one is incorrect
my_list = LinkedList()
my_list.add(400)
print(my_list.remove_from_tail())
my_list.add(300)
print(my_list.remove_from_tail())
my_list.add(200)
print(my_list.remove_from_tail())
my_list.add(100)
print(my_list.remove_from_tail())
#should be 400 300 200 100 but instead I got 400 400 300 200

#test case two works fine
fruit = LinkedList()
fruit.add('cherry')
fruit.add('banana')
fruit.add('apple')
last_one = fruit.remove_from_tail()
print('Removed:', last_one)#got"Removed: cherry"
print(fruit)#(apple, banana)

我不知道在除去400之后，cur = self.head和self.head应该指向300时，测试用例一失败的原因是什么。因此，当我返回cur时，不应打印出两个400。任何帮助将不胜感激。

Answer 1

您的代码很好，问题在于您每次添加后都调用remove_from_tail，但是您要执行的操作是调用add直到添加所有元素，然后调用{{1} }一对一，效果很好

remove_from_tail

这将输出

my_list = LinkedList()
my_list.add(400)
my_list.add(300)
my_list.add(200)
my_list.add(100)
print(my_list.remove_from_tail())
print(my_list.remove_from_tail())
print(my_list.remove_from_tail())
print(my_list.remove_from_tail())

此外，您的400 300 200 100函数有点奇怪，通常我们固定其头部，对于每个新元素，遍历列表的末尾并将其附加，但是在您的add中，您正在更新add上的元素并更新头部。

而您的head假定头部是固定的，并且我们要遍历到底，然后进行更新，这实际上是发生的

因此，运行添加4次后，列表为

remove_from_tail

并且正如您所期望的，尾巴也相应地从head 100 -> 200 -> 300 -> 400移至400

在您的100函数中，您还需要处理以下情况：列表中只剩下一个元素，并且一旦添加了原始测试用例就想删除它。

remove_from_tail

然后打印原始测试用例

def remove_from_tail(self):
    #If linked list is empty return None
    if self.head is None:
        return None
    #If only one element in list, return that element and set head to None
    elif self.head.next is None:
        cur = self.head
        self.head = None
        return cur

    #Otherwise iterate through the list
    prev = None
    cur = self.head
    while cur.next is not None:
        prev = cur
        cur = cur.next
    if prev:
        prev.next = None
    return cur

也是

如何删除并返回链表中的最后一项？

1 个答案: