对链表进行插入排序
中等
给定单个链表的头 head ,使用 插入排序 对链表进行排序,并返回 排序后链表的头 。
插入排序 算法的步骤:
插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
重复直到所有输入数据插入完为止。
下面是插入排序算法的一个图形示例。部分排序的列表(黑色)最初只包含列表中的第一个元素。每次迭代时,从输入数据中删除一个元素(红色),并就地插入已排序的列表中。
对链表进行插入排序。
示例 1:
输入: head = [4,2,1,3]
输出: [1,2,3,4]
示例 2:
输入: head = [-1,5,3,4,0]
输出: [-1,0,3,4,5]
题解
先把节点值存入数组
对数组值进行插入排序
把插入排序好的数组值形成链表返回
/**
* 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) {
ListNode dummy = head;
if(head == null || head.next == null){
return head;
}
int count = 0;
while (dummy != null){
count++;
dummy = dummy.next;
}
System.out.println(count);
int[] arr = new int[count];
int index = 0;
ListNode t = head;
while (t != null){
arr[index++] = t.val;
t = t.next;
}
// 对数组进行插入排序
for (int i = 1; i < arr.length; i++) {
int insertValue = arr[i];
int insertIndex = i - 1;
while (insertIndex >= 0 && arr[insertIndex] > insertValue){
arr[insertIndex + 1] = arr[insertIndex];
insertIndex--;
}
if(!(insertIndex+1 == i)){
arr[insertIndex+1] = insertValue;
}
}
System.out.println(Arrays.toString(arr));
// 构建链表
ListNode temp = new ListNode(0);
ListNode te = temp;
for (int i = 0; i < arr.length; i++) {
te.next = new ListNode(arr[i]);
te = te.next;
}
System.out.println(temp.next);
return temp.next;
}
}