148. Sort List

sort linked list using merge sort
# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def sortList(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        if head==None or head.next==None:
            return head
        #break the linked list to two equal parts, use prev node to break up the two list 
        prev,slow,fast=None,head,head
        while fast!=None and fast.next!=None:
            prev,slow,fast=slow,slow.next,fast.next.next
        prev.next=None
        sorted_l1=self.sortList(head)
        sorted_l2=self.sortList(slow)
        return self.mergeTwoLists(sorted_l1,sorted_l2)
        
    def mergeTwoLists(self,l1,l2):
        dummy=ListNode(-1)
        cur=dummy
        
        while l1!=None and l2!=None:
            if l1.val<=l2.val:
                cur.next,cur,l1=l1,l1,l1.next
            else:
                cur.next,cur,l2=l2,l2,l2.next
        if l1!=None:
            cur.next=l1
        if l2!=None:
            cur.next=l2
        return dummy.next

你可能感兴趣的:(148. Sort List)