题目:
给你一个链表的头节点 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-list-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
分类链表,题意其实就是把小于给定x的节点往前挪,这里会想到双指针,但是链表上不可能右指针向左移,这里使用双dummy node即可。用一个dummysamll记录小于x的node,用dummybig串联大于等于x的node,之后dummysamll后面接dummybig即可。
代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode dummysmall(0), dummybig(0);
ListNode* small = &dummysmall;
ListNode* big = &dummybig;
while (head) {
ListNode* tmp = head;
head = head->next;
tmp->next = nullptr;
if (tmp->val < x) {
small->next = tmp;
small = small->next;
} else {
big->next = tmp;
big = big->next;
}
}
small->next = dummybig.next;
return dummysmall.next;
}
};