给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
题解
- 使用两个头结点,lessHead, moreHead
- 小于接入lessHead,大于接入moreHead,最后将lessHead和moreHead链接起来就可以了
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode partition(ListNode head, int x) {
ListNode lessHead = new ListNode(0);
ListNode moreHead = new ListNode(0);
ListNode lessPtr = lessHead;
ListNode morePtr = moreHead;
while(head != null) {
if(head.val < x) {
lessPtr.next = head;
lessPtr = head;
} else {
morePtr.next = head;
morePtr = head;
}
head = head.next;
}
lessPtr.next = moreHead.next;
morePtr.next = null;
return lessHead.next;
}
}