思路:建立两个链表,大链表和小链表。大链表用于存储那些大于等于x的值,小链表用于存储小于x的值。然后把小链表拼接在大链表前面,并且令大链表的尾节点的next指针指向null。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode partition(ListNode head, int x) {
//建立指针指向大链表的头节点
ListNode bigHead = new ListNode(-1);
//建立指针指向大链表的尾节点
ListNode bigTail = bigHead;
//建立指针指向小链表的头节点
ListNode smallHead = new ListNode(-1);
//建立指针指向小链表的尾节点
ListNode smallTail = smallHead;
while(head != null){//head指针指向的节点不等于null就继续遍历
if(head.val < x){//如果head指针指向的节点值小于x
smallTail.next = head;//小链表的尾指针指向head指向的节点
smallTail = head;//尾指针移动到新的位置,即之前head指的位置
}else{
bigTail.next = head;//大链表的尾指针指向head指向的节点
bigTail = head;//尾指针移动到新的位置,即之前head指的位置
}
head = head.next;//小链表的指针或者大链表的指针移动到head的位置,head就继续向后移动
}
smallTail.next = bigHead.next;//循环结束以后,就把小链表的尾接到大链表的头上了
bigTail.next = null;//然后把大链表的尾置空了
return smallHead.next;
}
}
建立两个指针指向大链表的头节点和尾结点,因为大链表现在只有一个元素,所以头尾指针指向同一个节点。
while循环结束以后的局面是这样的,所以需要把大链表的尾部置空。