按特定值分隔链表

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

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

其实这道题直接的思路就可以做。首先要找到大于等于x的第一个值的前一个结点,作为插入位置的左边。然后再开始找小于x的节点做插入。注意每次插入后,插入位置的左边都要指向下一个元素,保证插入位置的右端不变。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* helperHead = new ListNode(INT_MIN, head);
        ListNode* curNode = helperHead;
        ListNode* insertLeft = helperHead;

        // 首先找到小于x的节点的插入位置
        while (curNode -> next != nullptr) {
            if (curNode -> next -> val >= x) {
                insertLeft = curNode;
                curNode = curNode -> next;
                break;
            }
            curNode = curNode -> next;
        }

        while (curNode -> next != nullptr) {
            if (curNode -> next -> val < x) {
                ListNode* tempNode = curNode -> next;
                curNode -> next = tempNode -> next;
                tempNode -> next = insertLeft -> next;
                insertLeft -> next = tempNode;
                insertLeft = insertLeft -> next;
                continue;
            }
            curNode = curNode -> next;
        }

        return helperHead -> next;

    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值