题目
插入排序的动画演示如上。从第一个元素开始,该链表可以被认为已经部分排序(用黑色表示)。
每次迭代时,从输入数据中移除一个元素(用红色表示),并原地将其插入到已排好序的链表中。
插入排序算法:
插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
重复直到所有输入数据插入完为止。
示例 1:
输入: 4->2->1->3
输出: 1->2->3->4
示例 2:
输入: -1->5->3->4->0
输出: -1->0->3->4->5
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def insertionSortList(self, head: ListNode) -> ListNode:
if not head or not head.next :
return head
#保存一个哑结点,并且哑结点下一个结点为头结点
dummyhead=ListNode(-1)
dummyhead.next=head
#暂时将排序好的结点设为头结点
lastNode=head
#维护cur为待插入的元素,设置为head.next避免和排序好的结点重复
cur=head.next
while cur:
#当lastNode结点不大于cur结点,lastNode结点后移
if lastNode.val<=cur.val:
lastNode=lastNode.next
else:
#当cur结点小于lastNode,寻找cur的插入位置
#首先找到cur之前的prev结点
prev=dummyhead
while prev.next.val<=cur.val:
prev=prev.next
#插入cur结点
lastNode.next=cur.next
cur.next=prev.next
#注意prev.next=cur
prev.next=cur
#cur结点指向lastNode的next
cur=lastNode.next
return dummyhead.next