LintCode 链表划分

题目描述：

给定一个单链表和数值x，划分链表使得所有小于x的节点排在大于等于x的节点之前。

你应该保留两部分内链表节点原有的相对顺序。

样例
给定链表 1->4->3->2->5->2->null，并且 x=3

返回 1->2->2->4->3->5->null

思路分析：

题意搞懂了就很简单了。

ac代码：

/**
 * Definition of ListNode
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 *     ListNode(int val) {
 *         this->val = val;
 *         this->next = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @param x: an integer
     * @return: a ListNode 
     */
    ListNode *partition(ListNode *head, int x) {
        // write your code here
        if(head==NULL) return NULL;

        ListNode *dummy=new ListNode(0);
        dummy->next=head;
        ListNode *temp=head;
        //如果要插入第一个位置的话，那么dummy结点可以帮助我们
        //避免pretemp是空结点而导致无法插入的错误。
        ListNode *pretemp=dummy;

        ListNode *cur=NULL;
        ListNode *precur=NULL;

        //找到第一个大于等于x的值作为分界线，存在temp中。
        while(temp!=NULL&&temp->val<x)
        {
            pretemp=temp;
            temp=temp->next;
        }

        if(temp!=NULL&&temp->next!=NULL)
        {
            precur=temp;
            cur=temp->next;
        }
        //找到分界线右侧小于x的值的结点，将其存在cur中。
        while(cur!=NULL)
        {
            while(cur!=NULL&&cur->val>=x)
            {
                precur=cur;
                cur=cur->next;
            }
            //如果找到的话将其插入到分界线以前的第一个位置
            if(cur!=NULL)
            {
                precur->next=cur->next;
                cur->next=temp;
                pretemp->next=cur;
                pretemp=cur;
                cur=precur->next;
            }
        }

        temp=dummy;
        dummy=dummy->next;
        free(temp);
        return dummy;
    }
};