问题描述:
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。你应当保留两个分区中每个节点的初始相对位置。
示例一:
输入:head = [1,4,3,2,5,2], x = 3 输出:[1,2,2,4,3,5]
示例二:
输入:head = [2,1], x = 2 输出:[1,2]
提示:
- 链表中节点的数目在范围
[0, 200]
内 -100 <= Node.val <= 100
-200 <= x <= 200
思路:
因为题目要求保留两个分区中每个节点的初始相对位置,所以分别使用两个链表来储存大于和小于x的结点,最后将两条链表连接起来,返回小于x的链表头节点的后一位结点即可。
代码:
/**
* 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) {
if(!head || x < -100 || x > 100){//对应题目给出的数据做出的限制
return head;
}
ListNode* small_x = new ListNode(0);//小于x的链表
ListNode* big_x = new ListNode(0);//大于x的链表
ListNode* ps = small_x;
ListNode* pb = big_x;
while(head){
if(head->val < x){
ps->next = head;
ps = ps->next;
}
else{
pb->next = head;
pb = pb->next;
}
head = head->next;
}
pb->next = nullptr;//大于x的链表结尾赋值为nullptr
ps->next = big_x->next;//将两条链表结合
return small_x->next;
}
};