86. 分隔链表

leetcode力扣刷题打卡

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

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

解题思路

1、遍历链表，用两个数组分别存储小于x的节点以及大于等于x的节点；
2、链表结尾一定要指向空指针，p->next = nullptr;

原代码##

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        vector<ListNode *> v1, v2;
        ListNode *p = new ListNode(-1);
        ListNode *ans = p;
        while (head) {
            if (head->val < x) v1.emplace_back(head);
            else v2.emplace_back(head);
            head = head->next;
        }
        for (int i = 0; i < v1.size(); ++i) {
            p->next = v1[i];
            p = p->next;
        }
        for (int j = 0; j < v2.size(); ++j) {
            p->next = v2[j];
            p = p->next;
        }
        p->next = nullptr;
        return ans->next;
    }
};