刷题 - 分治

面试经典 150 题 - 分治

148. 排序链表⭐️⭐️⭐️ - 快慢指针找中间节点 - 归并排序


伪代码:

  • 将链表拆分成两半,返回右半边头节点(左半边头节点就是原始链表头节点)
  • 对左边进行排序并返回左边头节点
  • 对右边进行排序返回右边头节点
  • 传入左右半边头节点,合并两个升序链表
cpp 复制代码
// bug1: 把 cur_a->val <= cur_b->val 写成 cur_a <= cur_b
// 优化1:使用快慢指针找到中间节点,而不是两次遍历

//  分治的思想: 链表排序 = 断开左右半部分连接,对左半部分进行排序 + 对右半部分进行排序 + 合并两个升序链表
//  返回 升序链表的头节点
class Solution {
public:
    // 返回中间节点并拆分链表
    ListNode* splitList(ListNode* head) {
        if (head == nullptr || head->next == nullptr) return nullptr;
        ListNode* slow = head;
        ListNode* fast = head;
        ListNode* prev = nullptr;
        while (fast && fast->next) {
            prev = slow;
            slow = slow->next;
            fast = fast->next->next;
        }
        if (prev) prev->next = nullptr; // 拆分链表
        return slow;
    }
    // 合并两个升序链表 (不使用 dummy_head)
    ListNode* mergeList(ListNode* head_a, ListNode* head_b) {
        if (!head_a) return head_b;
        if (!head_b) return head_a;
        ListNode* merged_head = nullptr;  // 合并后链表的头节点
        ListNode* cur = nullptr;          // 当前操作的节点
        // 初始化 merged_head 和 cur
        if (head_a->val <= head_b->val) {
            merged_head = head_a;
            head_a = head_a->next;
        } else {
            merged_head = head_b;
            head_b = head_b->next;
        }
        cur = merged_head;
        // 合并剩余节点
        while (head_a && head_b) {
            if (head_a->val <= head_b->val) {
                cur->next = head_a;
                head_a = head_a->next;
            } else {
                cur->next = head_b;
                head_b = head_b->next;
            }
            cur = cur->next;
        }
        cur->next = (head_a != nullptr) ? head_a : head_b; 
        return merged_head;
    }

    // 归并排序
    ListNode* sortList(ListNode* head) {
        if (head == nullptr || head->next == nullptr) return head;
        // 拆分链表
        ListNode* mid = splitList(head);
        // 递归排序前半部分和后半部分
        ListNode* left = sortList(head);
        ListNode* right = sortList(mid);
        // 合并排序后的两个链表
        return mergeList(left, right);
    }
};
相关推荐
qq_479875433 小时前
C++ std::Set<std::pair>
开发语言·c++
云知谷5 小时前
【C++基本功】C++适合做什么,哪些领域适合哪些领域不适合?
c语言·开发语言·c++·人工智能·团队开发
仰泳的熊猫6 小时前
LeetCode:785. 判断二分图
数据结构·c++·算法·leetcode
^Moon^6 小时前
CycloneDDS:跨主机多进程通信全解析
c++·分布式·dds
C_Liu_8 小时前
C++:list
开发语言·c++
my rainy days8 小时前
C++:友元
开发语言·c++·算法
鄃鳕9 小时前
python 字典 列表 类比c++【python】
c++·python
保持低旋律节奏10 小时前
C++——list链表
c++·链表·list
2401_8414956411 小时前
【数据结构】基于Floyd算法的最短路径求解
java·数据结构·c++·python·算法··floyd
纵有疾風起12 小时前
C++模版:模板初阶及STL简介
开发语言·c++·经验分享·开源