OJ题-链表分割

最新推荐文章于 2024-10-30 16:53:07 发布

WJ8871

最新推荐文章于 2024-10-30 16:53:07 发布

阅读量954

点赞数

分类专栏：数据结构与算法 c++ 文章标签：链表数据结构算法

本文链接：https://blog.csdn.net/weixin_46153828/article/details/123414486

版权

c++ 同时被 2 个专栏收录

34 篇文章 0 订阅

订阅专栏

数据结构与算法

21 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用C++实现将链表中的节点按值大小分成两部分，小于x的节点构成一个链表，大于等于x的节点构成另一个链表。通过创建两个临时链表分别存储这两个部分，然后尾插节点，最后链接两个链表完成操作。

摘要由CSDN通过智能技术生成

题目：

解题思路：创建两个链表，分别存放小于x的节点和大于等于x的节点，分别进行尾插

代码如下：

class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
        if(pHead == NULL)
            return NULL;
        
        struct ListNode* lessHead, *lessTail,*greaterHead, *greaterTail;
        //创建链表表头
        lessHead = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        greaterHead = greaterTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        struct ListNode* cur = pHead;
        while(cur)
        {
            //小于x的尾插到lessTail
            if(cur->val < x)
            {
                lessTail->next = cur;
                lessTail = lessTail->next;
            }
            //大于等于x的尾插到greaterTail
            else
            {
                greaterTail->next = cur;
                greaterTail = greaterTail->next;
            }
            cur = cur->next;
        }
        //链接两个链表
        lessTail->next = greaterHead->next;
        greaterTail->next = NULL;
        //获取表头
        pHead = lessHead->next;
        free(lessHead);
        free(greaterHead);
        
        return pHead;
    }
};