给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
把所有小于给定值的节点都移到前面,大于该值的节点顺序不变。将所有小于给定值的节点取出组成一个新的链表,此时原链表中剩余的节点的值都大于或等于给定值,只要将原链表直接接在新链表后即可。
public ListNode partition(ListNode head, int x) {
if(head == null) return head;
ListNode dummy = new ListNode(-1);
ListNode newDummy = new ListNode(-1);
dummy.next = head;
ListNode cur = dummy;
ListNode p = newDummy;
while(cur.next != null){
if(cur.next.val < x){
p.next = cur.next;
p = p.next;
cur.next = cur.next.next;
p.next = null;
}
}
p.next = dummy.next;
return newDummy.next;
}