链接
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;
}
};