1 /**
 2  * Definition for singly-linked list.
 3  * struct ListNode {
 4  *     int val;
 5  *     ListNode *next;
 6  *     ListNode(int x) : val(x), next(NULL) {}
 7  * };
 8  */
 9 class Solution {
10 public:
11     ListNode *partition(ListNode *head, int x) {
12         if (head == NULL) return NULL;
13 
14         ListNode *ans = new ListNode(0);
15         ans->next = head;
16         ListNode *first_bigger = head;
17         ListNode *pre = ans;
18         while (first_bigger != NULL && first_bigger->val < x) {
19             first_bigger = first_bigger->next;
20             pre = pre->next;
21         }
22         ListNode *cur = first_bigger, *tmp = pre;
23         while (cur != NULL) {
24             while (cur != NULL && cur->val >= x) {
25                 tmp = cur;
26                 cur = cur->next;            
27             }
28             if (cur == NULL) break;
29             tmp->next = cur->next;
30             cur->next = pre->next;
31             pre->next = cur;
32             pre = cur;
33             cur = tmp->next;
34         }
35         return ans->next;
36     }
37 };