题目描述(传送门)
给你一个链表和一个特定值 x ,请你对链表进行分隔,使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入:head = 1->4->3->2->5->2, x = 3
输出:1->2->2->4->3->5
解题思路
使用两个带有傀儡节点的辅助链表,minNode和 maxNode,分别保存小于x的节点和大于等于x的节点。
遍历完之后,将minNode最后一个节点连接在maxnode的next,将maxNode最后的节点置为null。返回minNode.next.
代码实现
public ListNode partition(ListNode head, int x) {
ListNode minNode = new ListNode(-1);
ListNode maxNode = new ListNode(-1);
ListNode minNodeHead = minNode;
ListNode maxNodeHead = maxNode;
while (head != null) {
if (head.val >= x) {
maxNodeHead.next = head;
maxNodeHead = maxNodeHead.next;
}
if (head.val < x) {
minNodeHead.next = head;
minNodeHead = minNodeHead.next;
}
head = head.next;
}
minNodeHead.next = maxNode.next;
maxNodeHead.next = null;
return minNode.next;
}