【LeetCode练习题】Partition List

 

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,

题目意思为：给定一个链表和一个数，把这个链表中比这个数小的数全放在比这个数大的数前面，而且两边要保持原来的顺序。

思路有两种。

一是先创建两个新的链表，一个全是比x小的数，一个全是比x大的数。然后合并这两个链表。

第一种方法比较容易理解，具体可以参考：http://blog.csdn.net/havenoidea/article/details/12758079

下面给出第二种方法的代码。

struct ListNode {
      int val;
      ListNode *next;
      ListNode(int x) : val(x), next(NULL) {}
 };

//用cur和pre指向大于等于x的第一个节点和上一个节点，为了考虑第一个节点就大于x的情况，增加一个dummy的头结点便于处理方便。
class Solution {
public:
    ListNode *partition(ListNode *head, int x) {
        if(head == NULL){
            return head;
        }
        ListNode *dummy = new ListNode(0);
        dummy->next = head;
        head = dummy;
        bool hasGreater = false;
        //找到第一个大于等于x的节点，用cur指向它
        ListNode *pre = head, *cur = head->next;
        while(cur!=NULL){
            if(cur->val < x){
                pre = pre->next;
                cur = cur->next;
            }
            else{
                hasGreater = true;
                break;
            }
        }
        //将cur节点之后所有的小于x的节点都挪到pre的后面，cur的前面。
        if(hasGreater){
            ListNode *pre2 = head, *cur2 = head->next;
            bool isBehind = false;
            while(cur2!=NULL){
                if(cur2->val == cur->val){
                    isBehind = true;
                }
                if(isBehind && cur2->val < x){
                    pre2->next = cur2->next;
                    pre->next = cur2;
                    cur2->next = cur;
                    cur2 = pre2->next;
                    pre = pre->next;
                }
                else{
                    pre2 = pre2->next;
                    cur2 = cur2->next;
                }
            }
        }
        //去掉dummy节点。
        head = head->next;
        delete dummy;
        return head;    
    }
};