分隔链表
作者:互联网
目录
题目描述
- 题目地址:https://leetcode.cn/problems/partition-list/
- 题目要求
给你一个链表的头节点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]
解题思路
- 只需维护两个链表small 和large 即可,将 small 链表尾节点指向large 链表的头节点即可完成对链表的分隔
- 同时设small和large节点指向当前链表的末尾节点,开始时smallHead=small,largeHead=large
- 定义
smallHead
和largeHead
为两个链表的哑节点,它们的 next 指针指向链表的头节点,目的是为了更方便地处理头节点为空的边界条件 - 遍历结束后,我们将 \textit{large}large 的 \textit{next}next 指针置空,这是因为当前节点复用的是原链表的节点,而其 \textit{next}next 指针可能指向一个小于 x的节点
解题代码
var partition = function(head, x) {
let small = new ListNode(0);
const smallHead = small;
let large = new ListNode(0);
const largeHead = large;
while (head !== null) { //边界处理
if (head.val < x) {
small.next = head;
small = small.next; //small 的next 指针指向该节点,
} else {
large.next = head;
large = large.next; //否则将large 的 next 指针指向该节点。
}
head = head.next;
}
large.next = null;
small.next = largeHead.next;
return smallHead.next;
};
标签:large,head,分隔,next,链表,small,节点 来源: https://www.cnblogs.com/xiayuxue/p/16592850.html