题目
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
解题
- 创建两个虚拟头结点存储两个链表
- 遍历链表分别将两个链表节点放到两个链表中
- 连接两个链表
class Solution {
public ListNode partition(ListNode head, int x) {
// 小于x的链表
ListNode dummyHead1 = new ListNode(0);
// 大于等于x的链表
ListNode dummyHead2 = new ListNode(0);
dummyHead1.next = head;
ListNode node1 = dummyHead1;
ListNode node2 = dummyHead2;
while (head != null){
if (head.val >= x) {
// 将大于等于x的节点放入链表dummyHead2中
node2.next = head;
node2 = node2.next;
} else {
node1.next = head;
node1 = node1.next;
}
head = head.next;
}
node1.next = dummyHead2.next;
node2.next = null;
return dummyHead1.next;
}
}