题目:Sort a linked list using insertion sort.
思路:题目要求我们用插入排序来实现链表排序。我们构建一个当前排好序的链表,然后维护一个变量,不断指向链表中的下一个节点。用变量cur表示当前要插入的节点,每次循环找到cur节点在当前排好序的链表中对应的位置,然后插入进去,然后指向原链表中下一个节点,继续进行插入过程,直到原链表的所有节点都完成,既经过N次迭代后,就得到了排序好的结果。
复杂度:时间O(N^2) 空间O(1)
Attention:
1. 维护一个当前排序好的链表,构建一个虚拟头结点。我们总是将原链表中的下一个节点,插入到新链表中的合适位置。
ListNode* dummyhead = new ListNode(0);
ListNode* pre = dummyhead;
//在当前排好的新链表中找到第一个大于cur->val的结点
while(pre->next != NULL && pre->next->val <= cur->val)
{
pre = pre->next;
}
//当前pre的next结点的值大于cur的值,将cur插入到pre后
cur->next = pre->next;
pre->next = cur;
2. 维护一个节点next,表示原链表中的下一个要插入的节点。经过一轮迭代后,我们将cur 置为next. 这样我们就不用根据长度来判断下一个节点了。
ListNode* next = cur->next; //维护链表的下一个结点
cur = next; //cur指向原链表的下一个节点
3. 每次我们都需要从排序好的新链表的表头重新循环,找到要插入节点的合适位置。
pre = dummyhead; //重置pre为新链表头开始
AC Code:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode *insertionSortList(ListNode *head) {
if(head == NULL || head->next == NULL) return head;
ListNode* dummyhead = new ListNode(0);
ListNode* pre = dummyhead;
ListNode* cur = head;
while(cur != NULL)
{
ListNode* next = cur->next; //维护链表的下一个结点
pre = dummyhead; //重置pre为新链表头开始
//在当前排好的新链表中找到第一个大于cur->val的结点
while(pre->next != NULL && pre->next->val <= cur->val)
{
pre = pre->next;
}
//当前pre的next结点的值大于cur的值,将cur插入到pre后
cur->next = pre->next;
pre->next = cur;
cur = next; //cur指向原链表的下一个节点
}
return dummyhead->next;
}
};
这道题,刚开始我也有考虑在原来的链表中进行插入操作,就和数组的操作有点像,因为我们要将当前元素插入已经排好序的链表的合适位置上,这样我们不仅需要比较大小,还需要保证pre节点不能跨过当前节点,这样就无法保证我们的插入时正确的。如{3,4,1}, 第一次迭代pre->next停在3, cur在4,我们并不能将4插到3前面。所以对于链表,在原链表的操作是比较复杂的。需要新建一个排序好的链表进行插入操作,可以避免很多判断。