给你一个链表的头节点 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]
代码如下:
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* small=new ListNode(0);//定义链表,按顺序存放链表中<x的值
ListNode* smallHead=small;
ListNode* large=new ListNode(0);//定义链表,按顺序存放链表中>x的值
ListNode* largeHead=large;
while(head!=NULL)
{
if(head->val<x)
{
small->next=head;//head->val<x,放入small链表中
small=small->next;//让small链表往后走
}
else
{
large->next=head;//head->val>x,放入large链表中
large=large->next;//让large链表往后走
}
head=head->next;//遍历head链表
}
large->next=NULL;//当前节点复用的是原链表的节点,而其next指针可能指向一个小于x的节点,需要切断这个引用
small->next=largeHead->next;//small->next指向large链表真正意义上的头节点
return smallHead->next;//返回排序好之后的头节点
}
};