链表分解的时候,勿忘断开原链表中的每个节点的 next 指针(以LeetCode86.分隔链表为例)

本文通过86.分隔链表问题探讨了链表操作中的陷阱,强调了处理链表next指针的重要性。作者指出,遍历链表时直接修改next指针可能导致无限循环,解决方案是在每次节点赋值后断开next,或者在结束时设置最后一个p2节点的next为nullptr。文章通过代码实例展示了正确处理方式。
摘要由CSDN通过智能技术生成

作者在刷力扣的时候,遇到了这个问题。当时想不明白为何会报错,画了好多遍图都感觉没问题,搜也搜不到,最终debug发现了问题所在,希望分享给跟我一样苦于链表next迫害的小白菜鸟。

题目(以86.分隔链表为例)

给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。(本题例为x=3)
你应当 保留 两个分区中每个节点的初始相对位置。

在这里插入图片描述

题目分析

其实就是以x为阈值,比x小的放在一个链表中,比x大的放在一个链表中,最终合并起来。有的同学可能跟刚开始的我一样,想的是遍历两遍链表:链表要建立一个概念,它不跟数组一样是放在那里一成不变的。当你第一次遍历的时候,是会让里面的节点也都修改了,因此对于链表问题,别想着先遍历一遍再遍历一遍。这个题只能说是分解成两个链表,最后头尾相接,因此需要定位两个虚拟头节点!!涉及到链表的分解及合并。下面直接po代码

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode*dummy1 = new ListNode(-1);
        ListNode*dummy2 = new ListNode(-2);
        ListNode* p1 = dummy1;
        ListNode* p2 = dummy2;
        while(head!=nullptr)
        {
            if(head->val < x) 
            {
                p1->next = head;
                p1 = p1->next;
            }
            else
            {
                p2->next = head;
                p2 = p2->next;
            }
            //head = head->next;BUG容易出
            //断开原链表中的每个节点的 next 指针
            ListNode* temp = head->next;
            head->next = nullptr;
            head = temp;
        }
        p1->next = dummy2->next;
        return dummy1->next;
    }
};

为何需要断开next?

head = head->next;这句代码细节没处理好。为啥这么说呢?head是正常往下走了,但是其中有个问题就是会陷入无限循环。以本题为例,由于14325的5在被p2连起来之后,其由于指向head,因此head->next也是被继承过来的,所以后面是2,而head直接等于head->next后判断为2后,就直接加到p1后面了。
因此现在的情况就是下面所示:

p1: 1-->2-->2*
p2:4-->3-->5-->2*

现在问题就是p1中的第一个2指向的第二个2和p2中的5指向的2是一个指针(地址),你不把他们俩合并还好说,一旦合并,就成了1-->2-->2-->4-->3-->5-->2|-->4-->3-->5-->2|-->4-->...看出来了吗?形成了一个环了!
因此在每步的赋值结束后,应当对next清除,以防止在最后的时候陷入这种无限循环。
即:

 ListNode* temp = head->next;
 head->next = nullptr;
 head = temp;

或者,如果不想每个都断开,其实在head往下走的时候,每个p1和p2的next都在同时进行着更新,因此只有p2的最后一个是存在问题的,因此也可以加一句p2->next = nullptr;来解决
。即:

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode*dummy1 = new ListNode(-1);
        ListNode*dummy2 = new ListNode(-2);
        ListNode* p1 = dummy1;
        ListNode* p2 = dummy2;
        while(head!=nullptr)
        {
            if(head->val < x) 
            {
                p1->next = head;
                p1 = p1->next;
            }
            else
            {
                p2->next = head;
                p2 = p2->next;
            }
            head = head->next;
        }
        p2->next = nullptr;
        p1->next = dummy2->next;
        return dummy1->next;
    }
};
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值