leetcode 147. 对链表进行插入排序 python

如果取出的节点小于已排序的最后一个节点(上一个节点),则从头开始寻找插入的位置
# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def insertionSortList(self, head: ListNode) -> ListNode:
        if not head:
            return head
        
        dummyHead = ListNode(0) #哑节点
        dummyHead.next = head
        lastSorted = head #已排序的最后一个节点
        curr = head.next #取出的节点

        while curr: #当cuur不是最后一个节点时继续循环
            if lastSorted.val <= curr.val: #如果取出的节点的值大于等于上一个节点的值,就不动
                lastSorted = lastSorted.next #上一个节点进入下一个节点
            else:
                prev = dummyHead 
                while prev.next.val <= curr.val: #寻找应该插入的位置
                    prev = prev.next
                lastSorted.next = curr.next #让已排序的最后一个节点的位置为取出的节点的位置
                curr.next = prev.next #插入
                prev.next = curr #插入
            curr = lastSorted.next #取出下一个节点
        
        return dummyHead.next #返回哑节点的下一个节点(即排序后的头节点)

你可能感兴趣的:(leetcode,链表,leetcode,python)