Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.
You should preserve the original relative order of the nodes in each of the two partitions.
For example,
Given
return
Given
1->4->3->2->5->2 and x = 3,return
1->2->2->4->3->5.
思路很简单,就是merge的逆过程,用两个dummy节点来表示小于x和大于等于x的两部分,然后将两部分链接起来,尤其要注意的地方是要显示的将当前链表的尾部指向null以保证安全。
public ListNode partition(ListNode head, int x) {
if(head == null || head.next == null) {
return head;
}
ListNode dum1 = new ListNode(0);
ListNode dum2 = new ListNode(0);
ListNode temp1 = dum1, temp2 = dum2;
while(head != null) {
if(head.val < x) {
temp1.next = head;
temp1 = temp1.next;
head = head.next;
temp1.next = null; // safe
}
else {
temp2.next = head;
temp2 = temp2.next;
head = head.next;
temp2.next = null; // safe
}
}
temp1.next = dum2.next;
return dum1.next;
}
没有评论:
发表评论