[Leetcode] 第148题 排序链表

 

一、题目描述

在 O(n log n) 时间复杂度和常数级空间复杂度下，对链表进行排序。

示例 1:

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

示例 2:

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

二、题目分析

1）采用快排的思想，以第一个节点为基准，分成左右两部分分别排序

2）因为是链表，所以用一个整数cnt来标记要进行排序的链表节点的个数，不能超过这个数目

三、代码实现

class Solution {
public:
    ListNode* sortList(ListNode* head) {
        int cnt = 0;
        ListNode *p = head;
        while (p) {
            ++cnt;
            p = p->next;
        }
        return quickSort(head, cnt);
    }
private:
    ListNode* quickSort(ListNode *head, int cnt) {
        //cnt记录要进行排序的节点个数
        if (!head || !head->next || cnt == 0 || cnt == 1)
            return head;
        int left_cnt = 0, right_cnt = 0, base = head->val;
        ListNode* cur = head->next, *pre = head, *p = head;
        for (int i = 1; i < cnt; ++i) {
            if (cur&&cur->val < base) {
                ++left_cnt;
                pre->next = cur->next;
                cur->next = head;
                head = cur;
                cur = pre->next;
            }
            else if (cur&&cur->val >= base) {
                ++right_cnt;
                pre = cur;
                cur = cur->next;
            }
        }
        ListNode* left_node = quickSort(head, left_cnt);
        ListNode* right_node = quickSort(p->next, right_cnt);
        p->next = right_node;
        head = left_node;
        return head;
    }
};