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

作者在刷力扣的时候,遇到了这个问题。当时想不明白为何会报错,画了好多遍图都感觉没问题,搜也搜不到,最终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;
    }
};
  • 8
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
本题要求合并两个有序链表。对于链表的每一个节点,由于链表是有序的,所以可以将两个链表节点按照大小顺序进行比较,然后逐个将较小的节点链接上去,最终得到一个新的有序链表。 使用Rust语言实现这个问题,需要首先定义一个链表节点的结构体,包含node值以及next指针。然后定义一个函数来合并两个有序链表,输入为两个链表的头节点指针,输出为新的有序链表的头节点指针。 在合并过程,首先需要判断两个链表的头节点哪一个较小,将较小的节点作为新链表的头节点,并将该节点的next指针指向递归调用合并函数的结果。递归结束条件为其一个链表为空,则将另一个链表直接链接到新链表上。 完整代码如下: ```rust // 定义链表节点结构体 #[derive(Debug)] struct ListNode { val: i32, next: Option<Box<ListNode>>, } impl ListNode { fn new(val: i32) -> Self { ListNode { val, next: None } } } // 合并两个有序链表 fn merge_two_lists(l1: Option<Box<ListNode>>, l2: Option<Box<ListNode>>) -> Option<Box<ListNode>> { match (l1, l2) { (None, None) => None, // 两个链表均为空 (l1, None) => l1, // 其一个链表为空,直接返回另一个链表 (None, l2) => l2, (Some(mut l1), Some(mut l2)) => { if l1.val < l2.val { l1.next = merge_two_lists(l1.next, Some(l2)); Some(l1) } else { l2.next = merge_two_lists(Some(l1), l2.next); Some(l2) } } } } // 测试代码 fn main() { let l1 = Some(Box::new(ListNode { val: 1, next: Some(Box::new(ListNode { val: 2, next: Some(Box::new(ListNode { val: 4, next: None, })), })), })); let l2 = Some(Box::new(ListNode { val: 1, next: Some(Box::new(ListNode { val: 3, next: Some(Box::new(ListNode { val: 4, next: None, })), })), })); let merged = merge_two_lists(l1, l2); println!("{:?}", merged); } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值