LeetCode-148

最新推荐文章于 2023-03-05 22:09:05 发布

楚仙子

最新推荐文章于 2023-03-05 22:09:05 发布

阅读量107

点赞数

分类专栏： LeetCode 文章标签：算法 leetcode 排序算法归并排序

本文链接：https://blog.csdn.net/weixin_43901733/article/details/109912640

版权

LeetCode 专栏收录该内容

42 篇文章 0 订阅

订阅专栏

148. 排序链表

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

Example

在这里插入图片描述

input
head = [4,2,1,3]
output
[1,2,3,4]

在这里插入图片描述

input
head = [-1,5,3,4,0]
output
[-1,0,3,4,5]

Note

在 $O (n l o g n)$ 时间复杂度和常数级空间复杂度下，对链表进行排序
链表中节点的数目在范围 $0, 5×10^4]$ 内
$-10^5 \leq Node.val \leq 10^5$

思路

在这里插入图片描述

代码如下

class Solution {
public:
    ListNode* sortList(ListNode* head) {
        ListNode dummyHead(0);
        dummyHead.next = head;
        auto p = head;
        int length = 0;
        while (p) {
            ++length;
            p = p->next;
        }
        
        for (int size = 1; size < length; size <<= 1) {
            auto cur = dummyHead.next;
            auto tail = &dummyHead;
            
            while (cur) {
                auto left = cur;
                auto right = cut(left, size); 
                cur = cut(right, size); 
                
                tail->next = merge(left, right);
                while (tail->next) {
                    tail = tail->next;
                }
            }
        }
        return dummyHead.next;
    }
    
    ListNode* cut(ListNode* head, int n) {
        auto p = head;
        while (--n && p) {
            p = p->next;
        }
        
        if (!p) return nullptr;
        
        auto next = p->next;
        p->next = nullptr;
        return next;
    }
    
    ListNode* merge(ListNode* l1, ListNode* l2) {
        ListNode dummyHead(0);
        auto p = &dummyHead;
        while (l1 && l2) {
            if (l1->val < l2->val) {
                p->next = l1;
                p = l1;
                l1 = l1->next;       
            } else {
                p->next = l2;
                p = l2;
                l2 = l2->next;
            }
        }
        p->next = l1 ? l1 : l2;
        return dummyHead.next;
    }
};