给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3 输出: 1->2->2->4->3->5
方法一:
要求我们划分链表,把所有小于给定值的节点都移到前面,大于该值的节点顺序不变,相当于一个局部排序的问题。那么可以想到的一种解法是首先找到第一个大于或等于给定值的节点,用题目中给的例子来说就是先找到4,然后再找小于3的值,每找到一个就将其取出置于4之前即可
方法二:
就是将所有小于给定值的节点取出组成一个新的链表,此时原链表中剩余的节点的值都大于或等于给定值,只要将原链表直接接在新链表后即可
/**
* 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) {
if (head==null) return head;
ListNode dummy = new ListNode(0);
ListNode newDummy = new ListNode(0);
dummy.next = head;
ListNode cur = dummy, 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;
} else {
cur = cur.next;
}
}
p.next = dummy.next;
return newDummy.next;
}
}