086. Partition List

 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 == nullptr) return head;       // 链表为空，则直接返回
13         
14         // 操作原始链表的准备工作
15         // 包括创建一个辅助结点指向链表的头结点
16         ListNode *pHead = new ListNode(-1);
17         pHead->next = head;
18         ListNode *pre = pHead;      // pre指向遍历到的结点的前驱结点
19         ListNode *cur = head;       // cur指向遍历到的当前结点（即待处理结点）
20         
21         // 创建指向大于等于x(>= x ) 结点的链表
22         // 用pNew指针指向该链表的头结点，pNew为辅助结点
23         ListNode *pNew = new ListNode(-1);
24         ListNode *ptrNew = pNew;
25         
26         // 对原始链表进行遍历
27         while (cur != nullptr)
28         {
29             if (cur->val >= x)    // 将原始链表中大于等于x（>= x）的结点分离出来，形成新的链表
30             {
31                 ListNode *pTemp = cur;
32                 pre->next = pTemp->next;
33                 cur = cur->next;
34                 
35                 ptrNew->next = pTemp;
36                 ptrNew = ptrNew->next;
37                 ptrNew->next = nullptr;
38             }
39             else    // 对于原始链表中小于x(< x)的结点，移动指针指向即可。
40             {
41                 pre = cur;
42                 cur = cur->next;
43             }
44         }
45         
46         // 将两部分连接起来
47         // 将原始链表中小于x(< x)的部分和原始链表中大于等于x(>= x)的部分连接起来
48         pre->next = pNew->next;
49         head = pHead->next;
50         
51         // 释放辅助结点内存空间
52         delete pHead;
53         delete pNew;
54         
55         return head;
56     }
57 };