面试题 02.04. 分割链表(LeetCode)

我们上点强度,如果要保留各分区中节点的初始相对位置,应该怎么办? 

此时想法就是,创建两个lesshead和morehead链表指针,把小于x的节点放在lesshead链表里,把大于等于x的节点放在morehead链表里,最后再将两个链表链接起来

但是这里只用指针的话,会遇到空链表的情况要分类讨论,比较麻烦,所以这里我们可以创建哨兵位,用带头链表会比较方便 

 

但如果你是这样写的话,可能会遇到以下报错,这是为什么呢? 

 

 这是因为,你创建的哨兵位中next没有初始化,所以要将其置为NULL

 

判断的时候,如果cur指向的值小于x,则链接到lesstail上,反之,则链接到moretail上。不过,此时要注意,链接完以后,应该及时将尾部节点next置为NULL,否则后续可能会出现环状链表 

 

链接完两个链表后,记得要把哨兵位的空间给释放掉 

 

最终代码如下: 

struct ListNode* partition(struct ListNode* head, int x)
{
    struct ListNode* lesshead , *lesstail, *morehead, *moretail;
    lesshead = lesstail = (struct ListNode*)malloc(sizeof(struct ListNode));
    lesshead->next = NULL;
    morehead = moretail = (struct ListNode*)malloc(sizeof(struct ListNode));
    morehead->next = NULL;
    struct ListNode* cur = head;
        
    while (cur)
    {
        if (cur->val < x)
        {
            lesstail->next = cur;
            lesstail = lesstail->next;
            cur = cur->next;
            lesstail->next = NULL;
        }
        else
        {
            moretail->next = cur;
            moretail = moretail->next;
            cur = cur->next;
            moretail->next = NULL;
        }
    }

    lesstail->next = morehead->next;
    free(morehead);

    head = lesshead->next;
    free(lesshead);

    return head;
}

 

  • 6
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值