面试算法-103-对链表进行插入排序

Winn~2024-03-26 8:40

题目

给定单个链表的头 head ，使用插入排序对链表进行排序，并返回排序后链表的头。

插入排序算法的步骤:

插入排序是迭代的，每次只移动一个元素，直到所有元素可以形成一个有序的输出列表。

每次迭代中，插入排序只从输入数据中移除一个待排序的元素，找到它在序列中适当的位置，并将其插入。

重复直到所有输入数据插入完为止。

下面是插入排序算法的一个图形示例。部分排序的列表(黑色)最初只包含列表中的第一个元素。每次迭代时，从输入数据中删除一个元素(红色)，并就地插入已排序的列表中。

对链表进行插入排序。

示例 1：

输入: head = [4,2,1,3]

输出: [1,2,3,4]

解

java 复制代码

class Solution {
    public ListNode insertionSortList(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }

        ListNode dummy = new ListNode(0);
        dummy.next = head;

        ListNode p1 = dummy;

        while (p1.next != null) {
            ListNode p2 = p1;
            int min = Integer.MAX_VALUE;
            ListNode minPre = null;
            while (p2.next != null) {
                if (p2.next.val < min) {
                    min = p2.next.val;
                    minPre = p2;
                }
                p2 = p2.next;
            }

            ListNode minNode = minPre.next;
            if (minNode != null) {
                minPre.next = minNode.next;

                ListNode temp = p1.next;
                p1.next = minNode;
                minNode.next = temp;
            }
            p1 = p1.next;
        }
        return dummy.next;
    }
}