147. 对链表进行插入排序
给定单个链表的头 head ,使用 插入排序 对链表进行排序,并返回排序后链表的头 。
插入排序 算法的步骤:
- 插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
- 每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
- 重复直到所有输入数据插入完为止。
下面是插入排序算法的一个图形示例。部分排序的列表(黑色)最初只包含列表中的第一个元素。每次迭代时,从输入数据中删除一个元素(红色),并就地插入已排序的列表中。
对链表进行插入排序。
示例 1:
输入: head = [4,2,1,3]
输出: [1,2,3,4]
示例 2:
输入: head = [-1,5,3,4,0]
输出: [-1,0,3,4,5]
提示:
- 列表中的节点数在
[1, 5000]
范围内 -5000 <= Node.val <= 5000
解法
选择待插入节点,从链表头部开始依次比较判断找到插入位置,将节点插入该位置,如此往复。
public ListNode insertionSortList(ListNode head) {
if (head == null || head.next == null) {
return head;
}
ListNode dummyHead = new ListNode(0, head);
//当前判断的节点
ListNode cur = head.next;
//下一轮判断的节点
ListNode newCur = cur.next;
ListNode pre = head;
while (cur != null && dummyHead.next != newCur) {
newCur = cur.next;
if (pre.val <= cur.val) {
//不用移动
pre = cur;
cur = cur.next;
} else {
//找到插入位置
ListNode tmp = dummyHead;
while (tmp.next.val < cur.val && tmp != newCur) {
tmp = tmp.next;
}
//移动节点到tmp后面
pre.next = cur.next;
cur.next = tmp.next;
tmp.next = cur;
cur = newCur;
}
}
return dummyHead.next;
}