给定单个链表的头 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
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode insertionSortList(ListNode head) {
if (head == null || head.next == null) return head;
ListNode dummy = new ListNode(-1, head);
//lastNode 最后一个排序好的节点, cur当前节点
ListNode lastNode = head, cur = head.next;
while (cur != null) {
//当当前值小于排序好的最后一个节点, 就要从头开始遍历,找到插入位置
if (cur.val < lastNode.val) {
ListNode pre = dummy;
while (pre.next.val <= cur.val) pre = pre.next;
//pre的下一个位置就是插入点
ListNode next = cur.next;
cur.next = pre.next;
pre.next = cur;
cur = next;
// System.out.println(lastNode.val + ">>" + cur.val);
} else { //插入到lastNode后面就行
lastNode.next = cur;
lastNode = lastNode.next;
cur = cur.next;
}
}
lastNode.next = null;
return dummy.next;
}
}