LeetCode | 148. Sort List

最新推荐文章于 2024-11-10 14:06:08 发布

1LOVESJohnny

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量2k

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法链表

本文链接：https://blog.csdn.net/iLOVEJohnny/article/details/108891486

版权

LeetCode 专栏收录该内容

285 篇文章 0 订阅

订阅专栏

该博客讨论了如何使用快速排序的分治策略来对链表进行排序，提出了一种在O(nlogn)时间复杂度和O(1)额外空间复杂度下对链表进行排序的方法。通过定义一个辅助函数交换节点值，并使用getPartition函数找到链表的分区点，然后递归地对分区前后两部分进行排序，最终实现整个链表的升序排列。示例展示了不同输入情况下排序后的链表结构。

摘要由CSDN通过智能技术生成

题目：

Given the head of a linked list, return the list after sorting it in ascending order.

Follow up: Can you sort the linked list in O(n logn) time and O(1) memory (i.e. constant space)?

Example 1:

Input: head = [4,2,1,3]
Output: [1,2,3,4]

Example 2:

Input: head = [-1,5,3,4,0]
Output: [-1,0,3,4,5]

Example 3:

Input: head = []
Output: []

Constraints:

The number of nodes in the list is in the range [0, 5 * 104].
-105 <= Node.val <= 105

代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    void swap(int& a, int& b) {
        int tmp = a;
        a = b;
        b = tmp;
        return;
    }
    ListNode* getPartition(ListNode* head, ListNode* end) {
        ListNode *key = head, *p = head, *cur = head->next;
        while(cur != end)
        {
            if(cur->val < key->val)
            {
                p = p->next;
                swap(cur->val, p->val);
            }
            cur = cur->next;
        }
        swap(key->val, p->val);
        return p;
    }
    void sort(ListNode* head, ListNode* end) {
        if(head == end)
            return;
        ListNode* partition = getPartition(head, end);
        sort(head, partition);
        sort(partition->next, end);
    }
    ListNode* sortList(ListNode* head) {
        if(head == NULL || head->next == NULL)
            return head;
        sort(head, NULL);
        return head;
    }
};