其他分享
首页 > 其他分享> > LeetCode/排序链表

LeetCode/排序链表

作者:互联网

1. 合并两个有序链表

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode* preHead = new ListNode(-1);
        //一定要使用头指针并复制一个副本,不然后面循环操作不统一,未合并完的也不好加上去
        ListNode* prev = preHead;
        while (l1&&l2) {
            if (l1->val < l2->val) {
                prev->next = l1;//链入pre
                l1 = l1->next;//指针后移
            } else {
                prev->next = l2;
                l2 = l2->next;
            }
            prev = prev->next;//pre指针后移
        }
        // 合并后 l1 和 l2 最多只有一个还未被合并完,我们直接将链表末尾指向未合并完的链表即可
        prev->next = l1 == nullptr ? l2 : l1;

        return preHead->next;
    }
//也可以使用递归
};

2. 合并k个有序链表

//归并
ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(!lists.size()) return {};
        return merge(lists,0,lists.size()-1);
    }

    ListNode* merge(vector<ListNode*>& lists,int i ,int j){
        if(i==j) return lists[i];//边界
        ListNode*left=merge(lists,i,(i+j)/2);//分治
        ListNode*right=merge(lists,(i+j)/2+1,j);//分治
        return mergeTwoLists(left,right);//归并
    }
//顺序合并
ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *ans = nullptr;
        for (size_t i = 0; i < lists.size(); ++i) {
            ans = mergeTwoLists(ans, lists[i]);
        }
        return ans;
    }

3. 排序链表(归并)

//主函数
ListNode* sortList(ListNode* head) {
        // 1、递归结束条件
        if (!head||!head->next)  return head;
        
        // 2、找到链表中间节点并断开链表(两个返回第一个)
        ListNode* midNode = middleNode(head);
        ListNode* rightHead = midNode->next;
        midNode->next = nullptr;

        ListNode* left = sortList(head);//分治
        ListNode* right = sortList(rightHead);//分治
        // 3、当前层业务操作(合并有序链表)
        return mergeTwoLists(left, right);//归并
    }
//找中间节点的前一个节点
ListNode* middleNode(ListNode* head) {
        if(!head||!head->next) return head;
        ListNode* slow = head;
        ListNode* fast = head->next->next;
        while (fast != NULL && fast->next != NULL) {
            slow = slow->next;
            fast = fast->next->next;
        }
        return slow;
    }

4. 排序链表(极简归并)

class Solution {
public:
    ListNode* sortList(ListNode* head) {
        if (!head || !head->next) return head;
        auto slow = head, fast = head;
        while (fast->next && fast->next->next)
            slow = slow->next, fast = fast->next->next;
        // 切链
        fast = slow->next, slow->next = nullptr;
        return merge(sortList(head), sortList(fast));
    }

private:
    ListNode* merge(ListNode* l1, ListNode* l2) {
        ListNode sub(0), *ptr = &sub;
        while (l1 && l2) {
            auto &node = l1->val < l2->val ? l1 : l2;
            ptr = ptr->next = node, node = node->next;
        }
        ptr->next = l1 ? l1 : l2;
        return sub.next;
    }
};

标签:head,ListNode,next,链表,l2,l1,return,排序,LeetCode
来源: https://www.cnblogs.com/929code/p/16518138.html