问题描述
编写程序以 x 为基准分割链表,使得所有小于 x 的节点排在大于或等于 x 的节点之前。如果链表中包含 x,x 只需出现在小于 x 的元素之后(如下所示)。分割元素 x 只需处于“右半部分”即可,其不需要被置于左右两部分之间。
示例:
输入: head = 3->5->8->5->10->2->1, x = 5
输出: 3->1->2->10->5->5->8
解决办法
class Solution {
/**
* 3->5->8->5->10->2->1, x = 5
* 本质上是将小于的结点放在链表的前面即可
*/
public ListNode partition(ListNode head, int x) {
if(head == null)
return null;
ListNode current = head;
//注意,这里的遍历,忽略了第一个结点。
//但是在将小于X结点插入到链表头部的过程中,原有第一个结点
//始终保持在“新插入结点的尾部“
while(current.next!=null){
if(current.next.val<x){
//将current.next往头部
ListNode tmp = current.next;
//删除currentNext结点
current.next = current.next.next;
//将currentNext结点放入头部
tmp.next= head;
head = tmp;
}else{
current = current.next;
}
}
return head;
}
}