给你一个链表和一个特定值 x ,请你对链表进行分隔,使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入:head = 1->4->3->2->5->2, x = 3
输出:1->2->2->4->3->5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-list
解题思路
思路清晰,只需要遍历一次链表,遇到比 X 小的节点加到单独一边,大于等于 x 的节点加到另外一边,最后小的那边的尾节点的 next 指向大的部分的头节点即可
代码
class Solution {
public ListNode partition(ListNode head, int x) {
if (head == null || head.next == null) {
return head;
}
ListNode smallHead = new ListNode(-1);
ListNode small = smallHead;
ListNode largeHead = new ListNode(-1);
ListNode large = largeHead;
while (head != null) {
if (head.val < x) {
small.next = head;
small = small.next;
} else {
large.next = head;
large = large.next;
}
head = head.next;
}
large.next = null; // 为了断开链表可能存在的环
small.next = largeHead.next;
return smallHead.next;
}
}