【牛客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;
    }
};

算法代码优化

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

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

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

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


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

相关推荐
码爸2 小时前
b树详细介绍与示例
数据结构·b树
leobertlan9 小时前
痛苦系列 | DSP-03 跨越单位圆:Z变换与零极点分析
算法·程序员
蛋蛋的就会顺顺的9 小时前
hot100 p0——栈,贪心算法,动态规划
java·前端·javascript·算法·leetcode·力扣
开发小能手12310 小时前
答案daanC语言
数据结构·算法
死磕音视频10 小时前
让小模型做选择题:限定候选项并读取 logits 概率的入门实践(附 Python 代码)
开发语言·python·大模型·文本分类·transformers
Navigator_Z11 小时前
LeetCode //C - 1291. Sequential Digits
c语言·算法·leetcode
WL学习笔记12 小时前
C & C++ 内存管理
c++·c++内存管理
pride.li12 小时前
第03章 Python 教程-基础语法:注释与变量
服务器·前端·后端·python·算法·计算机视觉
顽疲12 小时前
Java vue 养老系统源码详解:月账单 care_month_bill 与 Quartz 扣费 Job 实践
java·开发语言·vue.js·vue养老·springboot养老·java养老
Wang's Blog13 小时前
Java 项目实战: 交友项目 - 安装开发工具与环境准备
java·开发语言·交友