c语言-数据结构-链表分割

链表分割实际上是给定一个值,遍历链表把链表中小于该值的节点与大于该值的节点分开,一般是将小于该值的节点放到链表的前面部分,大于该值的节点放在链表的后面部分。

链表分割示意图如下:

思路: 首先创建两条带哨兵位节点的新链表并且用big、small指针维护,比8大的节点放在big链表,比8小的节点放在small链表。

两条新链表则需要创建两个哨兵位。哨兵位节点的作用顾名思义,仅仅是起到一个"站岗"的作用,也就是把要插入到新链表中的节点直接插到哨兵位节点的后面。

情况一:

遍历链表,发现比8大的节点将其放到big哨兵位节点的后面,如下图:

注意:这里不能直接让small和big指针进行遍历,因为如果这两个指针往后移动则找不到哨兵位节点的位置了,也就找不到链表的初始位置,还会面临内存泄漏的风险(因为哨兵位是malloc来的,没有哨兵位的位置就无法对其进行释放)。因此再定义两指针b_travel和s_travel代替移动,并且将cur指针移动至下一个节点。

情况二:

若发现比8小的节点将其放到small哨兵位节点的后面,如下图:

同理将cur指针至下一个节点, 每次拿下来一个节点都要更新b_travel和s_travel指针指向的位置。如此重复以上操作,直到cur指针指向NULL说明链表已遍历结束。

合并链表:

接下来是最关键的一步,就是将small链表和big链表合二为一,最后返回small哨兵位的下一个节点的地址,即合并后链表的头节点地址。

切记最后返回的是small哨兵位节点的下一个节点的地址,因为small和big开辟的哨兵位节点只是用于"站岗",最后链表链接完成后需要将这两个哨兵位释放。

铺垫了那么久接下来就是代码实现(测试代码包括链表的创建、打印以及链表分割功能):

cpp 复制代码
#include<stdio.h>
#include<stdlib.h>

typedef struct ListNode //创建节点结构体
{
    int data;//存放数据
    struct ListNode* next;//指向下一个节点
}ListNode;

//链表分割函数
ListNode* partition(ListNode* head, int x) {
    struct ListNode* small = NULL;//开始先创建4个指针,两两维护链表
    struct ListNode* big = NULL;
    struct ListNode* s_travel = NULL;//这两个指针用于遍历链表
    struct ListNode* b_travel = NULL;

    //开辟两个哨兵位节点
    s_travel = small = (struct ListNode*)malloc(sizeof(struct ListNode));
    b_travel = big = (struct ListNode*)malloc(sizeof(struct ListNode));
    
    struct ListNode* cur = head;//用cur指针代替head遍历链表
    while (cur)//cur为空跳出循环
    {
        if (cur->data < x)//小于x的节点放到small链表中
        {
            s_travel->next = cur;
            s_travel = s_travel->next;
        }
        else//大于x的节点放到big链表中
        {
            b_travel->next = cur;
            b_travel = b_travel->next;
        }
        cur = cur->next;//cur往后走
    }

    b_travel->next = NULL;//合并链表
    s_travel->next = big->next;

    struct ListNode* poi = small;//记住small哨兵位节点的位置
    small = small->next;//small指向哨兵位后一个节点,也就是链表的头节点
    free(big);//释放哨兵位
    free(poi);
    return small;//返回链表头节点位置
}

//打印链表函数
void Print(ListNode* phead)
{
    ListNode* cur = phead;
    while (cur)
    {
        printf("%d->", cur->data);
        cur = cur->next;
    }
    printf("NULL");//模拟链表最后指向的是NULL
}

int main()
{
    ListNode* n1 = (ListNode*)malloc(sizeof(ListNode));//创建5个节点,为了测试分割链表函数
    ListNode* n2 = (ListNode*)malloc(sizeof(ListNode));
    ListNode* n3 = (ListNode*)malloc(sizeof(ListNode));
    ListNode* n4 = (ListNode*)malloc(sizeof(ListNode));
    ListNode* n5 = (ListNode*)malloc(sizeof(ListNode));

    ListNode* plist = n1;//指向头节点的头指针plist
    n1->data = 12;//给每个节点都赋值
    n2->data = 6;
    n3->data = 23;
    n4->data = 18;
    n5->data = 2;

    n1->next = n2;//手动构建链表
    n2->next = n3;
    n3->next = n4;
    n4->next = n5;
    n5->next = NULL;

    Print(plist);//打印分割之前的链表

    ListNode* newhead = partition(plist, 8);//分割后返回新的头节点

    printf("\n分割后:");
    Print(newhead);//打印分割之后的链表

    
    return 0;
}

运行结果:

结语:

以上就是关于链表分割全部讲解与实现,其中对哨兵位节点合理的使用可以很方便解决某些链表问题,这种思路在链表问题中尤为重要,希望本文可以对你起到帮助。

相关推荐
湫ccc9 分钟前
《Python基础》之字符串格式化输出
开发语言·python
mqiqe1 小时前
Python MySQL通过Binlog 获取变更记录 恢复数据
开发语言·python·mysql
AttackingLin1 小时前
2024强网杯--babyheap house of apple2解法
linux·开发语言·python
Ysjt | 深1 小时前
C++多线程编程入门教程(优质版)
java·开发语言·jvm·c++
ephemerals__2 小时前
【c++丨STL】list模拟实现(附源码)
开发语言·c++·list
码农飞飞2 小时前
深入理解Rust的模式匹配
开发语言·后端·rust·模式匹配·解构·结构体和枚举
一个小坑货2 小时前
Rust 的简介
开发语言·后端·rust
湫ccc2 小时前
《Python基础》之基本数据类型
开发语言·python