- 题目:
给你一个链表的头节点 head 和一个特定值 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 || head.next == null)
return head;
ListNode sh = null;
ListNode st = null;
ListNode bh = null;
ListNode bt = null;
// 关键步骤:用next保存一下head的下一位,因为后面要把每个head的下一位设null,保证每次每次赋值head赋的都是单个节点
ListNode next = null;
while (head != null){
// 先把当前head的下一个节点暂存在next中
next = head.next;
// 把head下一个节点的指向改为null,防止后面赋值head的时候赋值一大串链表
head.next = null;
// head值小于x时,把head添加到sh链表中
if (head.val < x){
// sh==null说明小数链表还没添加过,就把头尾都设为head
if (sh == null){
sh = head;
st = head;
// 如果小数链表添加过,只需要把head赋给尾节点st的next指向,然后让st指向下一个节点
}else{
st.next = head;
st = st.next;
}
}else {
if (bh == null){
bh = head;
bt = head;
}else {
bt.next = head;
bt = bt.next;
}
}
// 最后利用暂存的next节点更新head
head = next;
}
// 判断一下st是否为空防止空指针
if (st != null)
st.next = bh;
return sh == null ? bh : sh;
}