链表快排

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了链表快排相关的知识,希望对你有一定的参考价值。

给定一个单向链表,在O(1)空间复杂度和O(nlogn)时间复杂度下进行排序

# -*- coding: utf-8 -*-
# @Time         : 2019-04-19 20:07
# @Author       : Jayce Wong
# @ProjectName  : job
# @FileName     : linkedListQuickSort.py
# @Blog         : https://blog.51cto.com/jayce1111
# @Github       : https://github.com/SysuJayce

# Definition for singly-linked list.
class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

def linkedListQuickSort(head):
    sort_helper(head, None)
    return head

def sort_helper(head, end):
    """
    快排本质上来讲就是选择一个主元,比主元小的值都放到主元左边(以升序为例),其他都放到右边
    因此不论是数组还是链表,快排的递归代码是一样的
    """
    # 由于我们这里判断了head是否和end相等来作为递归的出口,因此不必向数组那样判断下标起止点的大小
    # 来作为递归出口
    if head != end:
        pivot = quick_sort(head)
        sort_helper(head, pivot)
        sort_helper(pivot.next, end)

def quick_sort(head):
    # 由于在链表中我们不能像数组那样选择一个下标来作为主元位置,因此就选择待排序链表的第一个元素作
    # 为主元即可。
    fast, slow = head.next, head  # slow是用来标记比主元小的元素应在的位置,fast用来遍历
    while fast:
        # 当发现有元素比主元(head)的值小的时候,我们先确定这个元素应该在的节点,然后将当前节点
        # 的值和应该在的节点的值交换
        if fast.val < head.val:
            slow = slow.next
            slow.val, fast.val = fast.val, slow.val

        fast = fast.next

    # 最后将主元的值放到正确的节点上
    slow.val, head.val = head.val, slow.val
    # 返回主元所在的节点
    return slow

def main():
    head = ListNode(3)
    head.next = ListNode(4)
    head.next.next = ListNode(2)
    head.next.next.next = ListNode(5)

    head = linkedListQuickSort(head)
    while head:
        print(head.val)
        head = head.next

if __name__ == ‘__main__‘:
    main()

以上是关于链表快排的主要内容,如果未能解决你的问题,请参考以下文章

链表快排 & 基于链表的排序

算法学习——单链表快排

链表问题-----排序

链表排序中为什么归并比快排好?

链表排序中为什么归并比快排好?

链表的快速排序问题