LeetCode - 86. 分隔链表

描述

给你一个链表和一个特定值 x ，请你对链表进行分隔，使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例：

输入：head = 1->4->3->2->5->2, x = 3
输出：1->2->2->4->3->5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-list/

求解

说明：最开始直接 new 节点，后面改为了智能指针，当然，其实这儿没有必要

    class Solution {
    public:
        ListNode *partition(ListNode *head, int x) {
            if (head == nullptr || head->next == nullptr) {
                return head;
            }

//            ListNode *lessHead = new ListNode(0); // 小于x头结点
            std::shared_ptr<ListNode> lessHead = std::make_shared<ListNode>(ListNode(0)); // 小于x头结点
            ListNode *lh = lessHead.get();
//            ListNode *greaterEqualHead = new ListNode(0); // 大于等于x头结点
            auto greaterEqualHead = std::make_shared<ListNode>(0); // 大于等于x头结点
            ListNode *geh = greaterEqualHead.get();
            ListNode *p = head;
            while (p) {
                if (p->val < x) {
                    lh->next = p;
                    lh = lh->next;
                    p = p->next;
                    continue;
                }
                geh->next = p;
                geh = geh->next;
                p = p->next;
            }
            geh->next = nullptr; // 注意，大于等于段的数据最后一个结点的next应该赋值nullptr,表示链表结束
            lh->next = greaterEqualHead->next; // 小于段的数据最后一个结点应该链接到大于等于数据段的开始
            ListNode *res = lessHead->next;
//            delete lessHead;
//            delete greaterEqualHead;
            return res;
        }
    };