python中的链表实现的合并排序不起作用

时间:2016-01-19 10:26:18

标签: python sorting data-structures mergesort

有人可以帮我弄清楚这段代码出了什么问题吗?代码正在链接列表中打印第一个节点,而不是打印已排序的链接列表。

    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()

2 个答案:

答案 0 :(得分:2)

此响应适用于早期版本的代码。有关新版本代码的修复,请参阅我的新响应。

好的,看起来问题就是你从函数中返回链表的方式。在front_to_back_split中,您要分配给head1head2,但这些只是该功能的参数。即,它们是局部变量。分配给他们对调用者没有影响。

更好的方法是将head1head2作为参数消除,而只是将它们作为普通的局部变量。然后将其更改为返回head1head2,如下所示:

return head1, head2

然后,在merge_sort中,您不再需要分配ab。相反,你可以这样做:

a, b = front_to_back_split(head)

同样,merge_sort应该返回新的head,以便调用者可以使用它。否则,调用者无法确定新列表头是什么。

答案 1 :(得分:1)

好的,我已经调试了您的更新版本,现在可以使用了。有三个变化:

  1. merge_sort的顶部,只有return。将其更改为:

    return head
    
  2. merge_sort中,更改递归调用,以便他们更新ab,如下所示:

    a = merge_sort(a)
    b = merge_sort(b)
    
  3. 在您的主代码中,在对列表进行排序后,您需要一个带有新头的LinkedList才能打印它,因为llist1仍会指向旧头。你可以用这个:

    print "Sorted list"
    new_head = merge_sort(llist1.head)
    new_list = LinkedList()
    new_list.head = new_head
    new_list.print_list()