【leetcode】86 分隔链表（链表）

题目链接：https://leetcode-cn.com/problems/partition-list/

题目描述

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

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

示例:

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

思路

用两个链表分别存储有小于 x 的节点和大于或等于 x 的节点，最后将两个链表链接
注意可以用空节点作为链表头，最后返回第二个节点，方便处理第一个节点为空的情况

代码

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        if(!head || !head->next) return head;
        ListNode *lHead = new ListNode(-1);
        ListNode *rHead = new ListNode(-1);
        ListNode *p = lHead;    // 小于x的指针
        ListNode *q = rHead;    // 大于等于x的指针
        while (head){
            if (head->val < x){
                p->next = head;
                p = head;
            }
            else{
                q->next = head;
                q = head;
            }
            head = head->next;
        }
        q->next = nullptr;      // 注意将最后一个节点指向空节点
        p->next = rHead->next;
        return lHead->next;
    }
};