算法之链表分割链表

问题描述

编写程序以 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;
    }
}