给你一个链表的头节点 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
public ListNode partition(ListNode head, int x) {
// 初始化左半部分链表的头和尾节点
ListNode leftHead = null, leftTail = null;
// 初始化右半部分链表的头和尾节点
ListNode rightHead = null, rightTail = null;
// 用于保存下一个节点的指针
ListNode next = null;
// 遍历链表节点
while (head != null) {
// 保存当前节点的下一个节点
next = head.next;
// 断开当前节点与下一个节点的连接
head.next = null;
// 判断节点值与 x 的大小关系
if (head.val < x) {
if (leftHead == null) {
leftHead = head;
} else {
leftTail.next = head;
}
leftTail = head;
} else {
if (rightHead == null) {
rightHead = head;
} else {
rightTail.next = head;
}
rightTail = head;
}
// 移动到下一个节点
head = next;
}
// 如果左半部分链表为空,直接返回右半部分链表
if (leftHead == null) {
return rightHead;
}
// 将左半部分链表的尾节点与右半部分链表的头节点连接起来
leftTail.next = rightHead;
// 返回整合后的链表
return leftHead;
}