Python插入节点函数

Question

我很难理解如何正确修复插入节点函数。我没有收到任何错误，但我也没有显示该列表。

class Node: 
    def __init__(self, data): 
     self.data = data 
     self.next = None 

class Solution: 
    def display(self, head): 
     current = head 
     while current: 
      print current.data, 
      current = current.next 

    def insert(self, head, data): 
     self.head = head 
     new_node = Node(data) 
     if self.head is None: 
      self.head = new_node 
     else: 
      new_node.next = self.head 
      self.head = new_node 

mylist = Solution() 
T = int(input()) 
head = None 
for i in range(T): 
    data = int(input()) 
    head = mylist.insert(head, data)  
mylist.display(head) 

Answer 1

您的插入（）没有返回任何东西。因此，如果您为其分配返回值，那么head将为None。

Answer 2

我觉得这是你想要

class Node: 
    def __init__(self, data): 
     self.data = data 
     self.next = None 


class Solution: 
    def __init__(self): 
     self.head = None 

    def display(self): 
     current = self.head 
     while current: 
      print current.data, 
      current = current.next 

    def insert(self, data): 
     new_node = Node(data) 
     if self.head is None: 
      self.head = new_node 
     else: 
      new_node.next = self.head 
      self.head = new_node 


mylist = Solution() 
T = int(input()) 
for i in range(T): 
    data = int(input()) 
    mylist.insert(data) 
mylist.display()