【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();
         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;
     }
 };