【leetcode】86. 分隔链表（partition-list）（模拟）[中等]

链接

https://leetcode-cn.com/problems/partition-list/

耗时

解题：37 min
题解：12 min

题意

给你一个链表和一个特定值 x ，请你对链表进行分隔，使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

思路

添加哑节点作为头节点，然后找到第一个比 x 大的前一个节点的位置，之后从这个节点的位置开始向后遍历，如果遍历的当前节点元素小于 x 说明这个节点需要换到前面，利用 当前节点的前一个节点 和 第一个比 x 大的前一个节点 完成删除和插入，并将 第一个比 x 大的前一个节点的位置 往后移一位。

时间复杂度：$O(n)$

AC代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* dummy_head = new ListNode(-1);
        dummy_head->next = head;
        ListNode* bef_greater_x = dummy_head;
        while(bef_greater_x->next != NULL && bef_greater_x->next->val < x) 
            bef_greater_x = bef_greater_x->next;
        
        if(bef_greater_x->next == NULL) return head;
        
        ListNode* bef_lower_x = bef_greater_x->next;
        while(bef_lower_x != NULL && bef_lower_x->next != NULL) {
            if(bef_lower_x->next->val < x) {
                ListNode* lower_x = bef_lower_x->next;
                bef_lower_x->next = lower_x->next;
                lower_x->next = bef_greater_x->next;
                bef_greater_x->next = lower_x;
                bef_greater_x = lower_x;
                continue;
            }
            bef_lower_x = bef_lower_x->next;
        }
        
        return dummy_head->next;
    }
};