2. 解題思路
- 從頭至尾的遍歷自然維護着節點之間的相對位置
- 採用兩個鏈表分別記錄小於x和大於等於x的節點
3. 算法
- left_dummy和right_dummy分別表明兩個鏈表的啞節點
- 從頭開始遍歷鏈表,當節點val小於x時,更新left_dummy鏈表;不然,更新right_dummy鏈表
- 循環完成後,將兩個鏈表鏈接起來
4. 實現
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode left_dummy(-1);
ListNode right_dummy(-1);
ListNode *left_cur = &left_dummy;
ListNode *right_cur = &right_dummy;
for(ListNode *cur = head; cur != NULL; cur = cur->next){
if(cur->val < x){
left_cur->next = cur;
left_cur = cur;
}else{
right_cur->next = cur;
right_cur = cur;
}
}
left_cur->next = right_dummy.next;
right_cur->next = NULL;
return left_dummy.next;
}
};