leetcode算法

leetcode链表之排序

2022-03-30  本文已影响0人  小奚有话说

147、对链表进行插入排序

题目:

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

插入排序 算法的步骤:

对链表进行插入排序

示例1:

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

示例2:

输入: head = [-1,5,3,4,0]
输出: [-1,0,3,4,5]

思路:

由于是插入排序,可以模拟数组插入排序,记录已排序序列的最后一个值,如果当前值小于这个值,说明需要将当前值插入到已排序序列中,那么从头开始比较,找到可以插入值的位置进行插入。

class Solution:
    def insertionSortList(self, head: ListNode) -> ListNode:
        if not head or not head.next: return head
        vHead = ListNode(0, head)
        curSorted = head # 已排序的结点
        cur = head.next # 当前遍历到的结点
        while cur:
            # 排序值小于当前值,说明当前是有序的,让排序值后移即可
            if curSorted.val <= cur.val:
                curSorted = curSorted.next
            else:
                # 遍历找到可以插入的位置
                prev = vHead
                while prev.next.val <= cur.val:
                    prev = prev.next
                # 记录当前值的下一个指针,避免指针丢失
                curSorted.next = cur.next
                # 进行插入
                cur.next = prev.next
                prev.next = cur
            # 将当前指针恢复到之前循环的位置
            cur = curSorted.next
        return vHead.next

148、排序链表

题目

给你链表的头结点 head ,请将其按 升序 排列并返回 排序后的链表

示例1:

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

示例2:

输入:head = [-1,5,3,4,0]
输出:[-1,0,3,4,5]

示例3:

输入:head = []
输出:[]

思路

采用归并思想进行链表排序

class Solution:
        # 找到链表中间
    def getMedian(self, left, right):
        slow = fast = left
        while fast != right and fast.next != right:
            fast = fast.next.next
            slow = slow.next
        return slow
        # 合并链表
    def merge(self, l1, l2):
        cur = vHead = ListNode()
        while l1 and l2:
            if l1.val <= l2.val:
                cur.next = l1
                l1 = l1.next
            else:
                cur.next = l2
                l2 = l2.next
            cur = cur.next
        cur.next = l1 or l2
        return vHead.next
        # 采用归并排序
    def mergeSort(self, left, right):
        if not left: return left
        # 如果左节点的下一个节点指向右节点,说明当前只有一个结点,可以将后续结点置空,返回进行排序
        if left.next == right:
            left.next = None
            return left
        mid = self.getMedian(left, right)
        return self.merge(self.mergeSort(left, mid), self.mergeSort(mid, right))

    def sortList(self, head: Optional[ListNode]) -> Optional[ListNode]:
        if not head: return head
        return self.mergeSort(head, None)
上一篇 下一篇

猜你喜欢

热点阅读