86. 分隔链表 哨兵节点,两种实现

题目

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

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

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

解法1:哨兵节点,在原链表挑小数节点到辅助链表里

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        if (NULL == head || NULL == head->next)
            return head;
        
        ListNode small(0), big(0);
        ListNode *ptrSmall = &small, *ptrBig = &big;
        big.next = head;                               //在big链表里挑小数节点到small链表里
        while(ptrBig->next) {
            if (ptrBig->next->val >= x) {
                ptrBig = ptrBig->next;
            } else {
                ptrSmall->next = ptrBig->next;
                ptrSmall = ptrSmall->next;
                ptrBig->next = ptrBig->next->next;
            }
        }
        ptrSmall->next = big.next;

        return small.next;
    }
};

 解法2:直接处理原链表,小数节点放一个辅助链表,大数链表放一个辅助链表

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        if (NULL == head || NULL == head->next)
            return head;
        
        
        ListNode small(0), big(0);
        ListNode *s = &small, *b = &big;
        while (head) {                        //直接处理原链表
            if (head->val < x) {
                s->next = head;
                s = s->next;
            } else {
                b->next = head;
                b = b->next;
            }
            head = head->next;
        }

        s->next = big.next;
        b->next = NULL;                       //注意尾节点置空,否则可能会出现环

        return small.next;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值