给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
示例 1:
输入:head = [1,4,3,2,5,2], x = 3
输出:[1,2,2,4,3,5]
示例 2:
输入:head = [2,1], x = 2
输出:[1,2]
解题
使用四个变量解决此题,此题为小于x的放在前面,大于等于x放后面,链表顺序不能改变。
小于x的变量为小头smallHead,小尾smallTail, 大头确定之后,小尾依次向下移动即可,形成小于x的链表。
大于等于x的变量为大头bigHead,大尾bigTail,组合模式参考小链表。
大小新链表组合完成之后,将大链表的头部接到小链表之后 新链表完成。
public ListNode partition(ListNode head, int x) {
//申请组合大小两个新链表的变量
ListNode bigHead = null,bigTail = null;
ListNode smallHead = null, smallTail = null;
//原链表head不为空循环
while(head != null){
//将原链表拆分,head链表只有一个值。
//例如1->2->3->4->5, 拆成head = 1;
//next = 2->3->4->5。
ListNode next = head.next;
head.next = null;
//head比较
if(head.val >= x){
//新链表头为空将加入
if(bigHead == null){
bigHead = head;
}else{
//确定头部 尾部依次往下
bigTail.next = head;
}
//跳转尾部
bigTail = head;
}else{
if(smallHead == null){
smallHead = head;
}else{
smallTail.next = head;
}
smallTail = head;
}
//舍弃已经比较过的值,head从head.next开始。
head = next;
}
//没有小于x的链表 返回大于的链表即原始链表
if(smallHead == null){
return bigHead;
}
//将大头接入小链表尾部
smallTail.next = bigHead;
return smallHead;
}