【数据结构与算法】leetcode148.排序链表

链接:
题目:
给你链表的头结点 head ,请将其按 升序 排列并返回 排序后的链表 。
在这里插入图片描述


求解:
空间复杂度:O(1),时间复杂度:O(n(logn)) —>>>推荐用归并排序。
空间复杂度:O(n),时间复杂度:O(n(logn)) —>>>推荐用堆排序。
堆排序,好理解。

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 * };
 */
struct cmp{
    
    
    bool operator()(int a, int b){
    
    
        return a>b;
    }
};
class Solution {
    
    
public:
    /**
     * 
     * @param head ListNode类 the head node
     * @return ListNode类
     */
    ListNode* sortInList(ListNode* head) {
    
    
        // write code here
        priority_queue<int,vector<int>,cmp> heap;
        while(head){
    
    
            heap.push(head->val);
            head=head->next;
        }
        ListNode* dhead=new ListNode(-1);
        ListNode* cur=dhead;
        while(!heap.empty()){
    
    
            cur->next=new ListNode(heap.top());heap.pop();
            cur=cur->next;
        }
        return dhead->next;
    }
};

自底向上的归并排序

class Solution {
public:
    ListNode* sortList(ListNode* head) {
        if (head == nullptr) {
            return head;
        }
        int length = 0;
        ListNode* node = head;
        while (node != nullptr) {
            length++;
            node = node->next;
        }
        ListNode* dummyHead = new ListNode(0, head);
        for (int subLength = 1; subLength < length; subLength <<= 1) {
            ListNode* prev = dummyHead, *curr = dummyHead->next;
            while (curr != nullptr) {
                ListNode* head1 = curr;
                for (int i = 1; i < subLength && curr->next != nullptr; i++) {
                    curr = curr->next;
                }
                ListNode* head2 = curr->next;
                curr->next = nullptr;
                curr = head2;
                for (int i = 1; i < subLength && curr != nullptr && curr->next != nullptr; i++) {
                    curr = curr->next;
                }
                ListNode* next = nullptr;
                if (curr != nullptr) {
                    next = curr->next;
                    curr->next = nullptr;
                }
                ListNode* merged = merge(head1, head2);
                prev->next = merged;
                while (prev->next != nullptr) {
                    prev = prev->next;
                }
                curr = next;
            }
        }
        return dummyHead->next;
    }

    ListNode* merge(ListNode* head1, ListNode* head2) {
        ListNode* dummyHead = new ListNode(0);
        ListNode* temp = dummyHead, *temp1 = head1, *temp2 = head2;
        while (temp1 != nullptr && temp2 != nullptr) {
            if (temp1->val <= temp2->val) {
                temp->next = temp1;
                temp1 = temp1->next;
            } else {
                temp->next = temp2;
                temp2 = temp2->next;
            }
            temp = temp->next;
        }
        if (temp1 != nullptr) {
            temp->next = temp1;
        } else if (temp2 != nullptr) {
            temp->next = temp2;
        }
        return dummyHead->next;
    }
};


猜你喜欢

转载自blog.csdn.net/ryanji/article/details/127623937