题目:
给你一个链表的头节点 head
和一个特定值 x
,请你对链表进行分隔,使得所有 小于 x
的节点都出现在 大于或等于 x
的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
示例 1:
输入:head = [1,4,3,2,5,2], x = 3 输出:[1,2,2,4,3,5]
示例 2:
输入:head = [2,1], x = 2 输出:[1,2]
提示:
- 链表中节点的数目在范围
[0, 200]
内 -100 <= Node.val <= 100
-200 <= x <= 200
思路:
设置两个链表双指针分别存放小于x及大于等于x的结点,每个链表设置头结点和可遍历指向尾部的指针,设置指针遍历原始链表,判断所指向的值和指定值的大小,分别放入2个结点的后面,得到2个子链表后连接在一起
注意:每次放入分链表一个结点后要断开原链表的next结点,避免形成环
代码:
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* p = head;
ListNode * ptr1 = new ListNode(-1) ;
ListNode*p1 = ptr1;
ListNode *ptr2 = new ListNode(-1);
ListNode *p2 = ptr2;
while(p!=nullptr)
{
if(p->val<x)
{
p1->next = p;
p1 = p1->next;
}
else
{
p2->next = p;
p2 = p2->next;
}
ListNode* temp = p->next;
p->next = nullptr;
p = temp;
}
p1->next = ptr2->next;
return ptr1->next;
}
};