给定一个链表和一个值x,利用x把链表进行划分,比x小的都排列在比x大或者等于x的后面。链表原来元素的相对位置不变。
For example,
Given 1->4->3->2->5->2
and x = 3,
return 1->2->2->4->3->5
.
解答:想法是从前向后遍历,遇到第一个比x大的元素的时候停下,记录位置p1,同时始终有一个pre1记录p1之前的节点。这个时候从p1开始向后找,找到第一个比x小的元素节点,记录为p2,同时保持了p2的前驱节点。
再将p2插入到p1之前,同时对相应指针进行更新,进行下一轮的循环。
ListNode *partition(ListNode *head, int x) {
if(head == NULL || head->next == NULL) return head;
ListNode *p1 = head, *p2 = head,*dummy,*pre1,*pre2;
dummy = new ListNode(0);
dummy->next = head;
pre1 = dummy;
pre2 = dummy;
while(p1 != NULL && p2 != NULL)
{
if(p1->val < x){
p1 = p1->next;
pre1 = pre1 ->next;
p2 = p2->next;
pre2 = pre2->next;
}
else{
while(p2 != NULL)
{
if(p2 -> val < x)
{
//found the one to insert,insert it and move on
pre1->next = p2;
pre2->next = p2->next;
p2->next = p1;
p2 = pre2->next;
pre1 = pre1->next;
break;
}
else
{
p2 = p2->next;
pre2 = pre2->next;
}
}
}
}
return dummy->next;
}