给定单个链表的头 head
,使用 插入排序 对链表进行排序,并返回 排序后链表的头 。
插入排序 算法的步骤:
-
插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
-
每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
-
重复直到所有输入数据插入完为止。
下面是插入排序算法的一个图形示例。部分排序的列表(黑色)最初只包含列表中的第一个元素。每次迭代时,从输入数据中删除一个元素(红色),并就地插入已排序的列表中。
对链表进行插入排序。
示例 1:
输入: head = [4,2,1,3] 输出: [1,2,3,4]
示例 2:
输入: head = [-1,5,3,4,0] 输出: [-1,0,3,4,5]
代码如下:
class Solution {
public:
ListNode* insertionSortList(ListNode* head) {
if(head==nullptr)//如果链表为空,则不需要进行排序
{
return head;
}
//链表初始化操作
ListNode* dummyHead=new ListNode(0);//引入哑节点
dummyHead->next=head;//目的是在head之前插入节点
ListNode* lastSorted=head;//维护lastSorted为链表已经排好序的最后一个结点并初始化
ListNode* curr=head->next;//维护curr为待插入元素并初始化
while(curr!=nullptr)
{
if(lastSorted->val<=curr->val)//说明此时数字的顺序是正确的,不用进行插入操作
{
lastSorted=lastSorted->next;//只需要将lastSorted后移一位,curr变成新的lastSorted
}
else
{
ListNode *prev=dummyHead;//prev是插入节点curr位置的前一个结点
while(prev->next->val<=curr->val)//循环推出的条件是找到curr应该插入的位置
{
prev=prev->next;
}
//当链表中的数字顺序不正确时,将curr插入到prev的后面
lastSorted->next=curr->next;
curr->next=prev->next;
prev->next=curr;
}
curr=lastSorted->next;//一次插入完成之后,将curr更新为下一次要插入的元素
}
return dummyHead->next;//返回排好序的链表
}
};