牛客题目——链表分割

最新推荐文章于 2024-10-08 12:29:23 发布

热爱编程的小刘

最新推荐文章于 2024-10-08 12:29:23 发布

阅读量74

点赞数

分类专栏：牛客文章标签：链表数据结构

本文链接：https://blog.csdn.net/m0_74088266/article/details/130622930

版权

牛客专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目链接——链表分割

#include <cstdlib>
class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) 
    {
        ListNode* newtable1;
        ListNode* tailtable1;
        ListNode* newtable2;
        ListNode* tailtable2;
        newtable1 = tailtable1 = (ListNode*)malloc(sizeof(ListNode));
        newtable2 = tailtable2 = (ListNode*)malloc(sizeof(ListNode));
        while(pHead)
        {
            if(pHead->val < x)
            {
                tailtable1->next = pHead;
                tailtable1 = tailtable1->next;
            }
            else 
            {
                tailtable2->next = pHead;
                tailtable2 = tailtable2->next;
            }
            pHead = pHead->next;
        }
        tailtable1->next = newtable2->next;
        tailtable2->next = NULL;
        ListNode* n = newtable1->next;
        free(newtable1);
        free(newtable2);
        return n;
    }
};