给你一个链表的头节点 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路及方法
用两个链表分别记录,一个按序记录小于x的节点,一个按序记录大于等于x的节点。然后每个链表分别用一个指针进行拼接操作。
注意最后要把大链表指针.next置为null,因为可能指向原链表中的一个小于x的值。
/**
* 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 small = new ListNode();
ListNode pt_small = small;
ListNode large = new ListNode();
ListNode pt_large = large;
while (head != null) {
if (head.val < x) {
// 按序存储小于x节点
pt_small.next = head;
pt_small = pt_small.next;
} else {
// 按序存储大于等于x节点
pt_large.next = head;
pt_large = pt_large.next;
}
// 移动原链表
head = head.next;
}
// 拼接小大链表
pt_large.next = null;
pt_small.next = large.next;
return small.next;
}
}
结果如下:
网友评论