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

链接

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

耗时

解题:37 min
题解:12 min

题意

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

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

思路

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

时间复杂度: O ( n ) O(n) 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;
    }
};
上一篇:做Java开发7年,今年9个月时间圆梦饿了么P7


下一篇:Node-Red创建自定义节点