[LeetCode] 86. Partition List

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;
    }
};