1. 问题描述
对链表进行插入排序。
插入排序的动画演示如上。从第一个元素开始,该链表可以被认为已经部分排序(用黑色表示)。
每次迭代时,从输入数据中移除一个元素(用红色表示),并原地将其插入到已排好序的链表中。
插入排序算法:
- 插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
- 每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
- 重复直到所有输入数据插入完为止。
示例 1:
输入: 4->2->1->3 输出: 1->2->3->4
示例 2:
输入: -1->5->3->4->0 输出: -1->0->3->4->5
2. 代码实现
以前数组插入排序,是两重循环,第一重是遍历整个数组,第二个就是反向循环,找到前面第一个小于当前数的值,并插到它后面。但链表无法反向,所以第二重循环就干脆正向遍历一次,找到第一个大于当前结点值的结点,并插到它前面。
用p、q分别记录当前结点cur的前、后结点。用pre遍历一次找到要插入位置的前驱结点,若需要交换结点位置,则把cur插入到pre后面,再连接p和q;下一轮p不需要移动,q往后移一个。若不需要交换结点位置,下一轮p和q均需要后移一个。(q相当于记录下一轮的cur)
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode insertionSortList(ListNode head) {
if(head == null || head.next == null)
return head;
ListNode newhead = new ListNode(0);
newhead.next = head;
ListNode pre = newhead; //插入位置的前一个结点
ListNode cur = head; //当前要插入的结点
ListNode p = newhead, q = cur; //p、q分别表示当前结点cur的前、后结点
while(cur != null) {
q = cur.next;
pre = newhead;
while(pre.next != null && pre.next.val < cur.val) //找到插入位置
pre = pre.next;
if(pre.next == cur)
p = p.next;
else {
cur.next = pre.next;
pre.next = cur;
p.next = q;
}
cur = q;
}
return newhead.next;
}
}