86. Partition List (分割链表)
1. 题目翻译
给定一个单链表和一个值x,通过将链表中所有比x小的值移到所有大于等于x值的结点前将这些值分为两组。每一组内各个结点的顺序要和原链表相同。
例子:
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.
2. 解题方法
使用两个指针p和q,p指向第一个大于等于x的结点的前驱(即p->next是第一个大于等于x的值)。令q初始等于p,然后遍历寻找比x小的结点。因为要将比x小的结点移动,移动后该结点的前驱要指向该节点的后继,所以需要用q->next指向要移动的点而不是q。然后将q->next结点移动到p之前。
考虑到特殊情况,p本身就是头结点(即链表中第一个结点就比x大),为了不分类讨论,可以创建一个flag结点指向头结点,并让将初始化为flag。
3. 代码
//Runtime: 6ms
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(head==NULL)
return NULL;
ListNode* flag = new ListNode(0); //flag的值无所谓
flag->next = head;
ListNode* p =flag ;
//找到第一个比x大的结点的前驱
while(p->next&&p->next->val<x)
p = p->next;
ListNode* q = p;
while(q&&q->next){
//找到比x小的结点的前驱,并将比x小的结点换到前面
if(q->next->val < x){
ListNode* temp = q->next;
q->next = q->next->next;
temp->next = p->next;
p->next = temp;
p = p->next;
}else
q = q->next;
}
return flag->next;
}
};