有人可以帮我弄清楚这段代码出了什么问题吗?代码正在链接列表中打印第一个节点,而不是打印已排序的链接列表。
class LinkedList(object):
def __init__(self):
self.head = None
class Node(object):
def __init__(self,data):
self.data = data
self.next = None
def push(self, new_data):
new_node = self.Node(new_data)
new_node.next = self.head
self.head = new_node
def print_list(self):
temp = self.head
while(temp):
print temp.data
temp = temp.next
def merge_lists(head1, head2):
if(head1 is None):
return head2
if(head2 is None):
return head1
s = t= LinkedList.Node(None)
while(head1 and head2):
if(head1.data <= head2.data):
c= head1
head1 = head1.next
else:
c= head2
head2 = head2.next
t.next = c
t = t.next
t.next = head1 or head2
return s.next
def front_back_split(head):
if(head is None or head.next is None):
head1 = head
head2 = None
else:
slow = head
fast = head.next
while(fast != None):
fast = fast.next
if(fast!=None):
slow = slow.next
fast = fast.next
head1 = head
head2 = slow.next
slow.next = None
return head1, head2
def merge_sort(head):
if(head is None or head.next is None):
return
a,b = front_back_split(head)
merge_sort(a)
merge_sort(b)
new_head = merge_lists(a,b)
return new_head
if __name__=='__main__':
llist1 = LinkedList()
llist1.push(6)
llist1.push(7)
llist1.push(1)
llist1.push(4)
llist1.push(3)
llist1.push(8)
print "Sorted list"
new_head = merge_sort(llist1.head)
llist1.print_list()
答案 0 :(得分:2)
此响应适用于早期版本的代码。有关新版本代码的修复,请参阅我的新响应。
好的,看起来问题就是你从函数中返回链表的方式。在front_to_back_split
中,您要分配给head1
和head2
,但这些只是该功能的参数。即,它们是局部变量。分配给他们对调用者没有影响。
更好的方法是将head1
和head2
作为参数消除,而只是将它们作为普通的局部变量。然后将其更改为返回head1
和head2
,如下所示:
return head1, head2
然后,在merge_sort
中,您不再需要分配a
和b
。相反,你可以这样做:
a, b = front_to_back_split(head)
同样,merge_sort
应该返回新的head
,以便调用者可以使用它。否则,调用者无法确定新列表头是什么。
答案 1 :(得分:1)
好的,我已经调试了您的更新版本,现在可以使用了。有三个变化:
在merge_sort
的顶部,只有return
。将其更改为:
return head
在merge_sort
中,更改递归调用,以便他们更新a
和b
,如下所示:
a = merge_sort(a)
b = merge_sort(b)
在您的主代码中,在对列表进行排序后,您需要一个带有新头的LinkedList
才能打印它,因为llist1
仍会指向旧头。你可以用这个:
print "Sorted list"
new_head = merge_sort(llist1.head)
new_list = LinkedList()
new_list.head = new_head
new_list.print_list()