1.题目描述
给你一个链表的头节点
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
2.解题思路
因为要将所有小于x的结点提到大于等于x结点的前面,所以我们需要将整个链表分成两个部分,第一部分由小于x的结点组成,第二部分由大于等于x的结点组成,因为要保证两个分区中每个节点的初始相对位置,所以我们就在顺序遍历链表的过程中,遇到<x的结点就加入到表1,遇到>=x的结点就加入到表2,这样单个表中的顺序都是按原表中相对位置排列的,最终,让表1和表2连接起来,返回即可。
3.代码实现
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode partition(ListNode head, int x) {
ListNode l1 = new ListNode(-1);
ListNode l2 = new ListNode(-1);
ListNode p = l1, q = l2;
while (head != null) {
ListNode temp = head.next;
head.next = null;
if (head.val < x) {
p.next = head;
p = head;
} else {
q.next = head;
q = head;
}
head = temp;
}
p.next = l2.next;
return l1.next;
}
}