将所有小于x的结点排在其余结点之前,且不能改变原来的数据顺序

题目描述:现有一链表的头指针 ListNode* pHead,给一定值x,编写一段代码将所有小于x的结点排在其余结点之前,且不能改变原来的数据顺序,返回重新排列后的链表的头指针。

输入:2 4 9 3 2 1 6,x=5
输出:2 4 9 3 2 1 6

思路:遍历链表,依次拿链表的每个节点的val和x对比,开辟两个新的链表,将小于x的节点存入一个链表,将大于或者等于的节点存入另一个链表,最后将大的链表的头节点接入小的链表的尾部。
代码实现如下:

struct ListNode* partition(struct ListNode* pHead, int x)
    {
    //开辟两个新的链表
        struct ListNode* lessHead;
        struct ListNode* lessTail;
        struct ListNode* greaterHead;
        struct ListNode* greaterTail;
        lessHead = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        greaterHead = greaterTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        lessTail->next = NULL;
        greaterTail->next = NULL;

        struct ListNode* cur = pHead;
        while (cur)//比较val和x
        {
            if (cur->val < x)
            {
                lessTail->next = cur;
                lessTail = lessTail->next;
            }
            else
            {
                greaterTail->next = cur;
                greaterTail = greaterTail->next;
            }
            cur = cur->next;
        }
        
        //链接两个链表
        lessTail->next = greaterHead->next;

        pHead = lessHead->next;
        free(lessHead);
        lessHead = NULL;
        free(greaterHead);
        greaterHead = NULL;
        return pHead;
    }

但是其实上面的代码有一些案例可能无法实现,因为有些输入可能会造成链表成了一个圈。
例如;
输入:2 4 9 3 2 1 6 3,x=5
输出:2 4 9 3 3 2 1 6
如果输入这个案例,编译器就会报错,提示你内存不足,但是你只开辟了两个新的链表,怎么会内存过多呢?
这个案例和上一个案例相比区别在于输入最后多了一个3
因为在greater这个链表的最后一个节点是6,但是在原输入俩表中6->3,6的结点后面存入的是3,当while循环完成以后,6->next并没有置为NULL,所以这个6->next依然指向3,但是3是lessTail,将lessTail->next=greater->next时,会形成一个循环。
在这里插入图片描述

所以应该在while循环结束以后加一句
greaterTail->next=NULL;
更新后的代码如下:

struct ListNode* partition(struct ListNode* pHead, int x)
    {
        struct ListNode* lessHead;
        struct ListNode* lessTail;
        struct ListNode* greaterHead;
        struct ListNode* greaterTail;
        lessHead = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        greaterHead = greaterTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        lessTail->next = NULL;
        greaterTail->next = NULL;

        struct ListNode* cur = pHead;
        while (cur)
        {
            if (cur->val < x)
            {
                lessTail->next = cur;
                lessTail = lessTail->next;
            }
            else
            {
                greaterTail->next = cur;
                greaterTail = greaterTail->next;
            }
            cur = cur->next;
        }

        greaterTail->next = NULL;
        lessTail->next = greaterHead->next;

        pHead = lessHead->next;
        free(lessHead);
        lessHead = NULL;
        free(greaterHead);
        greaterHead = NULL;
        return pHead;
    }
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值