86. 分隔链表
题解
这题不难,事先准备两个链表,遍历题目给的链表,将小于x和大于等于x的两类数据分别放到准备好的链表中,然后两个链表拼接即可;
注意点:
链表在创建增加的时候,游走的那个指针在当前循环结束的时候不要指向下一个未知的结点,而是在下一次循环的时候去为下一个结点开辟空间,否则最后链表会多一个没有用的结点;
当无法在循环外确定第一个结点的时候,可以采用增加一个空的头结点的方式来辅助我们去在循环中创建增加。
代码:
public ListNode partition(ListNode head, int x) {
if (head == null) return null;
ListNode smallListNodeHead = new ListNode();
ListNode bigListNodeHead = new ListNode();
ListNode p = head, smallP = smallListNodeHead, bigP = bigListNodeHead;
while (p != null) {
ListNode newNode = new ListNode(p.val);
if (p.val < x) {
smallP.next = newNode;
smallP = newNode;
} else {
bigP.next = newNode;
bigP = newNode;
}
p = p.next;
}
smallP.next = bigListNodeHead.next;
return smallListNodeHead.next;
}