剑指Offer链表分割

题目描述

现有一链表的头指针 ListNode* pHead，给一定值x，编写一段代码将所有小于x的结点排在其余结点之前，且不能改变原来的数据顺序，返回重新排列后的链表的头指针。
分析
可以新new两个节点，一个连接比x值小的节点（before）一颗连接比x大的节点（after）。最后将两个新的链表一连接即可。
具体如下：

• cur（cur=head，遍历链表）
• bs(beforeStart小于x的节点的起始位置）
• be(beforeEnd小于x的节点的结束位置）
• as(afterStart大于x的节点的起始位置）
• ae(afterStart大于x的节点的结束位置）

如果cur.cal<x则将其连接到be后面（这里我们需要考虑第一次连接的情况，我们令bs=be=cur,大于也类似），否则将其连接到ae后面。然后接着遍历。

while(cur!=null)
        {
            if(cur.val<x)
            {
                if(bs==null)
                {
                    bs=cur;
                    be=cur;
                }else{
                     be.next=cur;
                     be=be.next;                    
                }               
            }else{
                if(as==null)
                {
                    as=cur;
                    ae=cur;
                }else{
                     ae.next=cur;
                     ae=ae.next;               
                }
               
            }
            cur=cur.next;
        }

还是之前的问题，我们需要重点考虑头和尾。
我们有一些特殊的情况：

• 如果链表的val全大于x（bs==null），则返回as。

• 连接后（be.next=as）后，如果as！=null，则需要将ae.next置为null，否则会出现死循环。

下面结合图片理解一下
开始

中间
尾部

代码如下：

public class Partition {
    public ListNode partition(ListNode pHead, int x) {
        
        // write code here      
        if(pHead==null)
        {
            return null;
        }
        ListNode cur=pHead;
        ListNode bs=null;
        ListNode be=null;
        ListNode as=null;
        ListNode ae=null;
        while(cur!=null)
        {
            if(cur.val<x)
            {
                if(bs==null)
                {
                    bs=cur;
                    be=cur;
                }else{
                     be.next=cur;
                     be=be.next;                    
                }               
            }else{
                if(as==null)
                {
                    as=cur;
                    ae=cur;
                }else{
                     ae.next=cur;
                     ae=ae.next;               
                }
               
            }
            cur=cur.next;
        }
        if(bs==null)
        {
            return as;
        }
       be.next=as;
        if(as!=null)
        {
            ae.next=null;
        }  
        
        return bs;   
    
    }
}