单链表---链表分割

将小于x的结点放在前面,大于等于x的结点放在后面,不改变结点相对位置,输出更改后的链表首结点。

cpp 复制代码
typedef struct ListNode
{
    ListNode* next;
    int val;
}ListNode;

思路:我们可以新创建两个链表指针,将小于x的结点全部链接到链表1,大于等于x的结点全部链接到链表2,然后将链表1尾结点链接链表2头结点。

方法一---不使用哨兵位

cpp 复制代码
    ListNode* cur = pHead;
    ListNode* head1 = NULL;
    ListNode* cur1 = head1;
    ListNode* head2 = NULL;
    ListNode* cur2 = head2;

当我们不使用哨兵位的时候,意味着,两个链表某一个为NULL的情况都属于特殊情况,不能解引用非法访问,因此在分割完的链接过程需要对cur1与cur2进行单独条件判断是否为空 。

当cur1为空说明了分割使所有结点都在链表2,那么cur2肯定不为空,因此将链表2尾结点存储的next置空然后返回链表2指针head2即可;cur1不为空,那么判断cur2是否为空,cur2为空则将链表1尾结点存储的next置空然后返回链表1指针head1即可,cur2不为空则正常链接两个链表然后将链表2尾结点存储的next置空,返回head1。

cpp 复制代码
    while (cur)
    {
        if (cur->val < x)
        {
            if (cur1 == NULL)
            {
                head1 = cur;
                cur1 = cur;
            }
            else
            {
                cur1->next = cur;
                cur1 = cur1->next;
            }
        }
        else
        {
            if (cur2 == NULL)
            {
                head2 = cur;
                cur2 = cur;
            }
            else
            {
                cur2->next = cur;
                cur2 = cur2->next;
            }
        }
        cur = cur->next;
    }

我们遍历原链表,然后进行结点的分配,由于没有哨兵位,我们需要对两个链表是否为空进行额外的判断如下:

cpp 复制代码
    if (cur1)
    {
        if (cur2)
        {
            cur2->next = NULL;
        }
        cur1->next = head2;
        return head1;
    }
    else
    {
        cur2->next = NULL;
        return head2;
    }

方法二---创建哨兵位

哨兵位的创建能够减少额外的判断,例如方法一中对两个链表指针是否为NULL的判断,由于哨兵位本身存在,所以head1与head2是不可能为NULL的,head1->next与head2->next才指向了真正有效的结点。那么我们就malloc两块结点空间作为两个哨兵位,然后进行操作:

cpp 复制代码
    ListNode* head1, * cur1, * head2, * cur2;
    head1 = cur1 = (ListNode*)malloc(sizeof(ListNode));
    head2 = cur2 = (ListNode*)malloc(sizeof(ListNode));
    ListNode* cur = pHead;

一般而言malloc开辟,需要判断是否开辟成功,请注意,同时最后需要释放空间,避免内存泄漏。

那么cur的遍历过程就会简单很多,不需要判断head1与head2是否为空了。

cpp 复制代码
    while (cur)
    {
        if (cur->val < x)
        {
            cur1->next = cur;
            cur1 = cur1->next;
        }
        else
        {
            cur2->next = cur;
            cur2 = cur2->next;
        }
        cur = cur->next;
    }

然后链表1尾结点next链接链表2头结点head2->next,(不是head2,head2是哨兵位!)

最后将cur2->next置空。

但是,哨兵位的方法,最终需要释放空间,因此我们需要先将需要返回的head1->next赋给我们的一个指针变量tmp,然后free掉head1与head2,返回tmp:

cpp 复制代码
    ListNode* tmp = head1->next;
    free(head1);
    free(head2);
    return tmp;
相关推荐
颜酱6 分钟前
图结构完全解析:从基础概念到遍历实现
javascript·后端·算法
QiZhang | UESTC8 分钟前
学习日记day76
学习
久邦科技16 分钟前
20个免费电子书下载网站,实现电子书自由(2025持续更新)
学习
m0_7369191018 分钟前
C++代码风格检查工具
开发语言·c++·算法
yugi98783820 分钟前
基于MATLAB强化学习的单智能体与多智能体路径规划算法
算法·matlab
Gain_chance30 分钟前
34-学习笔记尚硅谷数仓搭建-DWS层最近一日汇总表建表语句汇总
数据仓库·hive·笔记·学习·datagrip
DuHz39 分钟前
超宽带脉冲无线电(Ultra Wideband Impulse Radio, UWB)简介
论文阅读·算法·汽车·信息与通信·信号处理
Polaris北极星少女1 小时前
TRSV优化2
算法
Gain_chance2 小时前
36-学习笔记尚硅谷数仓搭建-DWS层数据装载脚本
大数据·数据仓库·笔记·学习
代码游侠2 小时前
C语言核心概念复习——网络协议与TCP/IP
linux·运维·服务器·网络·算法