algorithm-base/animation-simulation/链表篇/leetcode147对链表进行插入排序.md
2021-07-17 22:28:06 +08:00

10 KiB
Raw Blame History

如果阅读时,发现错误,或者动画不可以显示的问题可以添加我微信好友 tan45du_one ,备注 github + 题目 + 问题 向我反馈

感谢支持,该仓库会一直维护,希望对各位有一丢丢帮助。

另外希望手机阅读的同学可以来我的 公众号:袁厨的算法小屋 两个平台同步,想要和题友一起刷题,互相监督的同学,可以在我的小屋点击刷题小队进入。

有的老哥说链表的排序搞不明白,让我写一下,一不小心给忘记了,今天咱们就安排上。没有学过数据结构的同学可以先看下这个文章:

【绘图解析】链表详解

另外大家如果忘记了【动画模拟】插入排序【动画模拟】归并排序思想的话,可以先复习一下,不然这两道题目会看得云里雾里。

147. 对链表进行插入排序

示例 1

输入: 4->2->1->3 输出: 1->2->3->4

示例 2

输入: -1->5->3->4->0 输出: -1->0->3->4->5

下面咱们先来看个动画回忆一下插入排序的具体执行步骤,然后我们再来说下面的题目。

直接插入排序

我们的指针在数组时,可以随意的前后移动,将指针指向值和新元素的值比较后,将新元素插入到合适的位置。

我们知道链表查询元素的时间复杂度为O(n),我们只能够通过遍历链表查询元素。

那么我们怎么才能将新元素放到合适的位置呢?见下图。

此时我们不能通过移动绿色指针来寻找 5 的合适位置,那么我们应该怎么做呢?见下图。

当我们发现绿色指针的值大于新元素时7 > 5我们则可以定义一个新指针让其从哑节点开始遍历直到找到新元素5的位置4 和 7 之间),然后再将新元素插入即可。

我们是通过下面这行代码来找到插入位置的, temphead 则代表我们的紫色指针。

 while (temphead.next.val <= pre.val) {
      temphead = temphead.next;   
 } 

下面我们再来看动画模拟具体过程。

注:为了更好的表达算法思想,让过程更流畅,特省略了指针的移动细节,直接插入到合适位置,后面会详细说明插入操作的具体步骤。

我们通过上面的动画知道了大致过程,那么我们的是如何将新元素插入到指定位置的呢?

见下图。

插入排序

我们想要将 3 插入到 2 和 4 的中间,此时我们三个指针分别指向 243。

我们共分 4 步,来完成这个操作,见下图。

我们来把上图中的最后结果来捋一下看看最后结果是不是完成了插入操作。

这样我们就完成了链表的插入排序。下面我们来看下题目代码吧。

题目代码

Java Code:

class Solution {
    public ListNode insertionSortList(ListNode head) {
        if (head == null && head.next == null) {
            return head;
        }
        //哑节点
        ListNode dummyNode = new ListNode(-1);
        dummyNode.next = head;
        //pre负责指向新元素last 负责指向新元素的前一元素
        //判断是否需要执行插入操作
        ListNode pre = head.next;
        ListNode last = head;
        while (pre != null) {
            //不需要插入到合适位置,则继续往下移动
            if (last.val <= pre.val) {
                pre = pre.next;
                last = last.next;
                continue;
            }               
            //开始出发,查找新元素的合适位置
            ListNode temphead = dummyNode;
            while (temphead.next.val <= pre.val) {
                temphead = temphead.next;
            }           
            //此时我们已经找到了合适位置,我们需要进行插入,大家可以画一画
            last.next = pre.next;
            pre.next = temphead.next;
            temphead.next = pre;
            //继续往下移动
            pre = last.next;
        }
        return dummyNode.next;
    }
}

C++ Code:

class Solution {
public:
    ListNode* insertionSortList(ListNode* head) {
        if (head == nullptr && head->next == nullptr) {
            return head;
        }
        //哑节点
        ListNode * dummyNode = new ListNode(-1);
        dummyNode->next = head;
        //pre负责指向新元素last 负责指向新元素的前一元素
        //判断是否需要执行插入操作
        ListNode * pre = head->next;
        ListNode * last = head;
        while (pre != nullptr) {
            //不需要插入到合适位置,则继续往下移动
            if (last->val <= pre->val) {
                pre = pre->next;
                last = last->next;
                continue;
            }               
            //开始出发,查找新元素的合适位置
            ListNode * temphead = dummyNode;
            while (temphead->next->val <= pre->val) {
                temphead = temphead->next;
            }           
            //此时我们已经找到了合适位置,我们需要进行插入,大家可以画一画
            last->next = pre->next;
            pre->next = temphead->next;
            temphead->next = pre;
            //继续往下移动
            pre = last->next;
        }
        return dummyNode->next;
    }
};

JS Code:

var insertionSortList = function(head) {
    if (head === null || head.next === null) return head;
    //哑节点
    let dummyNode = new ListNode(-1, head);
    let pre = head.next;
    //pre负责指向新元素last 负责指向新元素的前一元素
    //判断是否需要执行插入操作
    let last = head;
    while (pre) {
        //不需要插入到合适位置,则继续往下移动
        if (last.val <= pre.val) {
            last = last.next;
            pre = pre.next;
            continue;
        }
        //开始出发,查找新元素的合适位置
        let tempHead = dummyNode;
        while (tempHead.next.val <= pre.val) {
            tempHead = tempHead.next;
        }
        //此时我们已经找到了合适位置,我们需要进行插入,大家可以画一画
        last.next = pre.next;
        pre.next = tempHead.next;
        tempHead.next = pre;
        //继续往下移动
        pre = last.next;
    }
    return dummyNode.next;
};

Python Code:

class Solution:
    def insertionSortList(self, head: ListNode) -> ListNode:
        if head is None or head.next is None:
            return head
        # 哑节点
        dummyNode = ListNode(-1, head)
        # pre负责指向新元素last 负责指向新元素的前一元素
        # 判断是否需要执行插入操作
        pre = head.next
        last = head
        while pre is not None:
            # 不需要插入到合适位置,则继续往下移动
            if last.val <= pre.val:
                pre = pre.next
                last = last.next
                continue
            # 开始出发,查找新元素的合适位置
            temphead = dummyNode
            while temphead.next.val <= pre.val:
                temphead = temphead.next
            # 此时我们已经找到了合适位置,我们需要进行插入,大家可以画一画
            last.next = pre.next
            pre.next = temphead.next
            temphead.next = pre
            # 继续往下移动
            pre = last.next
        return dummyNode.next

Swift Code

class Solution {
    func insertionSortList(_ head: ListNode?) -> ListNode? {
        if head == nil && head?.next == nil {
            return head
        }
        //哑节点
        var dummyNode = ListNode(-1)
        dummyNode.next = head
        //pre负责指向新元素last 负责指向新元素的前一元素
        //判断是否需要执行插入操作
        var pre = head?.next
        var last = head
        while pre != nil {
            //不需要插入到合适位置,则继续往下移动
            if last!.val <= pre!.val {
                pre = pre?.next
                last = last?.next
                continue
            }
            //开始出发,查找新元素的合适位置
            var temphead = dummyNode
            while temphead.next!.val <= pre!.val {
                temphead = temphead.next!
            }
            //此时我们已经找到了合适位置,我们需要进行插入,大家可以画一画
            last?.next = pre?.next
            pre?.next = temphead.next
            temphead.next = pre
            //继续往下移动
            pre = last?.next
        }
        return dummyNode.next
    }
}