【leetcode】链表分割

大家好,我是苏貝,本篇博客带大家刷题,如果你觉得我写的还不错的话,可以给我一个赞👍吗,感谢❤️


目录

  • [方法1. 不用哨兵位](#方法1. 不用哨兵位)
  • [方法2. 用哨兵位](#方法2. 用哨兵位)

点击查看题目

思路:

将链表分为2个链表list1和list2,list1是小于x的所有节点,list2是剩余的节点。最后将list1的最后一个节点指向list2的第一个节点

1

方法1. 不用哨兵位

c 复制代码
#include <csignal>
class Partition {
  public:
    ListNode* partition(ListNode* pHead, int x) {
        ListNode* list1 = nullptr;
        ListNode* i1 = nullptr;
        ListNode* list2 = nullptr;
        ListNode* i2 = nullptr;
        ListNode* cur = pHead;

        while (cur) {
            if (cur->val < x) {
                if (i1 == nullptr) {
                    i1 = list1 = cur;
                } else {
                    i1->next = cur;
                    i1 = i1->next;
                }
            } else {
                if (i2 == nullptr) {
                    i2 = list2 = cur;
                } else {
                    i2->next = cur;
                    i2 = i2->next;
                }
            }
            cur = cur->next;
        }
        if(list1==nullptr&&list2==nullptr)
            return nullptr;
        if(list1==nullptr&&list2!=nullptr)
            return list2;

        i1->next=list2;
        if(list2==nullptr)
            return list1;
        i2->next=nullptr;
        return list1;
    }
};

2

方法2. 用哨兵位

这种方法就不需要分多钟情况进行讨论,记得最后将开辟的空间释放掉,这是一个好习惯

c 复制代码
#include <csignal>
class Partition {
  public:
    ListNode* partition(ListNode* pHead, int x) {
        ListNode* list1,*i1,*list2,*i2,*cur;
        cur=pHead;
        list1=i1=(ListNode*)malloc(sizeof(ListNode));
        list2=i2=(ListNode*)malloc(sizeof(ListNode));
        while(cur)
        {
            if(cur->val<x){
                i1->next=cur;
                i1=i1->next;
            }
            else{
                i2->next=cur;
                i2=i2->next;
            }
            cur=cur->next;
        }
        i1->next=list2->next;
        i2->next=nullptr;
        pHead=list1->next;
        free(list1);
        free(list2);
        return pHead;
    }
};

好了,那么本篇博客就到此结束了,如果你觉得本篇博客对你有些帮助,可以给个大大的赞👍吗,感谢看到这里,我们下篇博客见❤️

相关推荐
HanhahnaH21 分钟前
各数据结构操作的时间复杂度汇总
数据结构·算法
Yzzz-F33 分钟前
CF2023D
c++·算法·dp
小小帅呀42 分钟前
学习 VLA 第6天:SWIN VIT算法原理以及复现
学习·算法
猎头南楼1 小时前
杭州算法工程师,偏大模型应用落地
算法
间歇性努力持续性发呆的野生快乐选手2 小时前
二分模板(整型,浮点型)
数据结构·c++·算法
evans在进步2 小时前
LeetCode 189 轮转数组:三次反转为什么能实现右旋?
数据结构·算法·leetcode
豆沙沙包?2 小时前
函数重载/类和对象(P14-P60)
前端·算法
lingchen19062 小时前
用MATLAB r2010b和r2016b求解微分方程组
开发语言·算法·matlab
无定义_3 小时前
Dijkstra——单源最短路径
算法·图论