086. 分隔链表

题目链接

一、题目描述

(一) 题目

给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。你应当保留两个分区中每个节点的初始相对位置

(二) 示例

示例 1:

输入:head = [1,4,3,2,5,2], x = 3

输出:[1,2,2,4,3,5]

示例 2:

输入:head = [2,1], x = 2

输出:[1,2]

(三) 提示

  • 链表中节点的数目在范围[0, 200]
  • -100 <= Node.val <= 100
  • -200 <= x <= 200

二、代码实现

(一) 迭代

1. 解题思路

这个题不是很难,只要理解题意加上构造两个子链即可完成,链表而不是数组,构建子链不增加空间复杂度。勇敢地构造子链,无需考虑节点交换。

图解如下:

通过图解,可以了解迭代的大致流程,但是我们在 Java 代码实现时,还需要维护两个对应的指针,用来调整它的 next 指针。

为什么需要这个指针而不是直接使用构造的节点呢?原因是我们在不断的遍历中,指针会不断的下移,当我们合并完成所有的节点时,此时的指针是在最后,这时我们在获取合并后的链表就比较麻烦了。

而用指针代替节点去进行 next 指针的移动,当返回链表时,只需返回节点的 next 即可。

为什么指针节点增加后,节点链表也会变动呢?这涉及到引用问题,这里直接引用拷贝的是堆中的内存地址,也就是说指针和对应节点就是一个对象,详情见我的另一篇文章《深拷贝和浅拷贝》

还有一个注意事项,遍历结束后,我们将 large 的 next 指针置空,这是因为当前节点复用的是原链表的节点,而其 next 指针可能指向一个小于 x 的节点,我们需要切断这个引用,否则可能会出现环链表。

代码实现如下:

class Solution {
    public ListNode partition(ListNode head, int x) {
        //构造两个子链表
        ListNode small = new ListNode(-1);
        ListNode large = new ListNode(-1);

        //构造子链表对应的指针,方便
        ListNode smallPointer = small;
        ListNode largePointer = large;

        while(head != null){
            if(head.val < x){
                smallPointer.next = head;	//追加节点
                smallPointer = smallPointer.next;	//指针后移
            }else{
                largePointer.next = head;	//追加节点
                largePointer = largePointer.next;	//指针后移
            }
            head = head.next;	//指针后移
        }

        //当前节点复用的是原链表的节点,很可能后面还指向一个小于x的节点
        largePointer.next = null;	
        smallPointer.next = large.next;		//链表拼接
        return small.next;
    }
}

2. 复杂度分析

时间复杂度

O(n),其中 n 是链表的长度,需要遍历链表一次。

空间复杂度

O(1),定义的变量使用的常数大小空间。

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我理解了您的问题。下面是C++代码实现: ```c++ #include <iostream> #include <vector> using namespace std; struct ListNode { int val; ListNode *next; ListNode(int x) : val(x), next(NULL) {} }; vector<ListNode*> splitListToParts(ListNode* head, int k) { vector<ListNode*> res(k, nullptr); int len = 0; for (ListNode* p = head; p; p = p->next) ++len; int n = len / k, r = len % k; ListNode* p = head, *prev = nullptr; for (int i = 0; p && i < k; ++i, --r) { res[i] = p; for (int j = 0; j < n + (r > 0); ++j) { prev = p; p = p->next; } prev->next = nullptr; } return res; } int main() { // 创建链表 ListNode* head = new ListNode(1); head->next = new ListNode(2); head->next->next = new ListNode(3); head->next->next->next = new ListNode(4); head->next->next->next->next = new ListNode(5); // 分隔链表 vector<ListNode*> parts = splitListToParts(head, 3); for (int i = 0; i < parts.size(); ++i) { cout << "Part " << i + 1 << ": "; ListNode* p = parts[i]; while (p) { cout << p->val << " "; p = p->next; } cout << endl; } return 0; } ``` 这段代码,我们使用了 `vector` 来存储分隔后的链表。首先计算链表长度 `len`,然后计算每个部分的长度 `n = len / k` 和余数 `r = len % k`。接下来我们遍历链表,将每个部分的头结点存储到 `vector` ,然后将链表断开,重复上述步骤,最后返回 `vector` 即可。 运行结果如下: ``` Part 1: 1 2 Part 2: 3 4 Part 3: 5 ``` 这里我们将长度尽可能相等的策略应用到了每个部分,因此部分 1 和 2 的长度相等,部分 3 的长度比它们短 1。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值