以前学c++的时候经常做这种题,但由于长时间没有用,现在已经忘得差不多了
题目:
对链表进行插入排序。
插入排序的动画演示如上。从第一个元素开始,该链表可以被认为已经部分排序(用黑色表示)。
每次迭代时,从输入数据中移除一个元素(用红色表示),并原地将其插入到已排好序的链表中。
插入排序算法:
插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
重复直到所有输入数据插入完为止。
示例 1:
输入: 4->2->1->3
输出: 1->2->3->4
示例 2:输入: -1->5->3->4->0
输出: -1->0->3->4->5
思路:
参考了官方思路:https://leetcode-cn.com/problems/insertion-sort-list/solution/dui-lian-biao-jin-xing-cha-ru-pai-xu-by-leetcode-s/
1. 首先判断给定的列表是否为空,若为空则直接返回。
2. 创建哑节点dummyHead,令dummyHead.next = head(这是在链表的题中经常用到的一种方式)
3. 维护 lastSorted 为链表的已排序部分的最后一个节点,初始时 lastSorted = head。
4. 维护 curr 为待插入的元素,初始时 curr = head.next。
5. 比较 lastSorted 和 curr 的节点值。
若 lastSorted.val <= curr.val,说明 curr 应该位于 lastSorted 之后,将 lastSorted 后移一位,curr 变成新的 lastSorted。
否则,从链表的头节点开始往后遍历链表中的节点,寻找插入 curr 的位置。令 prev 为插入 curr 的位置的前一个节点,进行如下操作,完成对 curr 的插入:
lastSorted.next = curr.next
curr.next = prev.next
prev.next = curr
6. 令 curr = lastSorted.next,此时 curr 为下一个待插入的元素。7. 重复第 5 步和第 6 步,直到 curr 变成空,排序结束。
8. 返回 dummyHead.next,为排序后的链表的头节点。
代码实现:
/**
* Definition for singly-linked list.
* function ListNode(val) {
* this.val = val;
* this.next = null;
* }
*/
/**
* @param {ListNode} head
* @return {ListNode}
*/
var insertionSortList = function(head) {
if(head===null) return head;
const dummyHead = new ListNode(0);
dummyHead.next = head;
let lastSorted = head, curr = head.next;
while(curr !== null) {
if(lastSorted.val <= curr.val) {
lastSorted = lastSorted.next;
} else {
let 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;
};