问题描述:
给你一个链表的头节点 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
代码:
struct ListNode* partition(struct ListNode* head,int x){
struct ListNode* small=malloc(sizeof(struct ListNode));
struct ListNode* smallhead=small;
struct ListNode* large=malloc(sizeof(struct ListNode));
struct ListNode* largehead=large;
while(head){
if(head->val<x){
small->next=head;
small=small->next;
}
else{
large->next=head;
large=large->next;
}
head=head->next;//整个循环结束后,small在最后一个小于x的结点处,large在最后一个大于x的结点处;
}
small->next=largehead->next;
large->next=NULL;//largehead在第一个大于x的结点处,smallhead在第一个小于x的结点处
return smallhead->next;
}