【牛客CM11】链表分割

刷爆LeetCode系列

牛客CM11:

github地址

有梦想的电信狗

前言

本文用C++实现牛客CM11题


题目描述

题目链接 :https://www.nowcoder.com/practice/0e27e0b064de4eacac178676ef9c9d70?tpId=8&&tqId=11004&rp=2&ru=/activity/oj&qru=/ta/cracking-the-coding-interview/question-ranking


题目与思路分析

目标分析:

  1. 编写代码,给定链表的头指针 pHead,以给定值x为基准 ,将链表分割成两部分,所有小于x的结点排在大于或等于x的结点之前

  2. 不能改变原来数据的顺序

  3. 返回 分割后的新链表的头指针

  4. 要求 :时间复杂度为O(n),空间复杂度为O(1)

思路 :创建两个哨兵位的头结点 guardLess 和 guardGreater,遍历链表,小于 x 的结点尾插 到 guardLess,大于 x 的结点尾插 到 guardGreater,最终再将两个链表连接起来,尾插的时候注意记录tail结点

操作:

  • 空链表检查 :if (pHead == nullptr) return nullptr;

  • 创建两个哨兵位的头结点:

cpp 复制代码
ListNode* guardLess = new ListNode(-1);
ListNode* guardGreater = new ListNode(-1);
  • 分别创建两个链表的尾结点,方便尾插时无需频繁找尾:
cpp 复制代码
ListNode* lessTail = guardLess, *greaterTail = guardGreater
  • curNode为移动指针,用于遍历链表:

    • 小于 x 的结点尾插到 guardLess 链表中,同时尾指针 lessTail 向后移动
    • 大于等于 x 的结点尾插到 guardGreater 链表中,同时尾指针 greaterTail 向后移动
    • 每次循环中curNode指针向后移动 : curNode = curNode->next
  • 连接两个新链表 : lessTail->next = guardGreater->next

  • 最后一个结点的 next 指针需要置空,防止链表带环 :greaterTail->next = nullptr

  • 保存新链表的头结点 : pHead = guardLess->next

  • 释放手动开辟的哨兵位头结点:

cpp 复制代码
  delete guardGreater;
  delete guardLess;
  • 最终返回新链表的头结点 : return pHead

代码实现

cpp 复制代码
class Partition {
  public:
    ListNode* partition(ListNode* pHead, int x) {
        if (pHead == nullptr)
            return nullptr;
        ListNode* guardLess = new ListNode(-1);
        ListNode* guardGreater = new ListNode(-1);

        ListNode* lessTail = guardLess, *greaterTail = guardGreater;
        ListNode* curNode = pHead;
        while (curNode) {
            if (curNode->val < x) {
                lessTail->next = curNode;
                lessTail = lessTail->next;
            }
            else{
                greaterTail->next = curNode;
                greaterTail = greaterTail->next;
            }
            curNode = curNode->next;
        }
        // 链接两个链表
        lessTail->next = guardGreater->next;
        
        // 最后一个结点的 next 指针需要置空,防止链表带环
        greaterTail->next = nullptr;    
		// 释放哨兵位的头结点
        pHead = guardLess->next;
        delete guardGreater;
        delete guardLess;

        return pHead;
    }
};

算法代码优化

  • 以上代码和思路已足够精简,无需优化

以上就是本文的所有内容了,如果觉得文章对你有帮助,欢迎 点赞⭐收藏 支持!如有疑问或建议,请在评论区留言交流,我们一起进步

分享到此结束啦
一键三连,好运连连!

你的每一次互动,都是对作者最大的鼓励!


征程尚未结束,让我们在广阔的世界里继续前行! 🚀

相关推荐
晓蛋3 天前
c语言指的是什么意思
c语言·编译器·编程开发·集成开发环境·程序实例
倒头就睡的小比特3 天前
算法竞赛C++常用的STL
c++·算法
weilx12343 天前
C++笔记-文件IO-<fcntl.h>
c++
小羊没烦恼!3 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
傲世仙尊3 天前
目录即文件-Ext文件系统收尾篇
linux·c语言
牵猫散步的鱼儿3 天前
重载、重写(覆盖)、重定义区别
c语言
伞伞悦读3 天前
【第38期】Python 模块与包详解:import、from、模块搜索路径、包结构和 __init__
开发语言·python
猎头南楼3 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
phltxy3 天前
C 语言指针:从内存地址到灵活的数据访问
c语言
Smileyqp沛沛3 天前
前端?C++ ?较大差异基础罗列
c++·基础·前端转c++