86. 分隔链表

最新推荐文章于 2024-11-07 13:20:34 发布

cx_cs

最新推荐文章于 2024-11-07 13:20:34 发布

阅读量46

点赞数

分类专栏：链表文章标签：链表数据结构

本文链接：https://blog.csdn.net/cx_cs/article/details/126414598

版权

链表专栏收录该内容

24 篇文章 0 订阅

订阅专栏

该代码实现了一个函数，将给定的单链表按特定值x进行划分，所有小于x的节点组成一个子链表，所有大于等于x的节点组成另一个子链表。通过创建两个头节点small和large，分别连接小于x和大于等于x的节点，最后将small后的链表连接到large的头节点上，返回small的新头节点。

摘要由CSDN通过智能技术生成

要求：快排操作
思路：不用管前驱后继，新建两个头节点连两个链表

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* small = new ListNode(0);
        ListNode* smallHead = small;
        ListNode* large = new ListNode(0);
        ListNode* largeHead = large;
        while (head != nullptr) {
            if (head->val < x) {
                small->next = head;
                small = small->next;
            } else {
                large->next = head;
                large = large->next;
            }
            head = head->next;
        }
        large->next = nullptr;
        small->next = largeHead->next;
        return smallHead->next;
    }
};