cc150 --链表分割

题目描述

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前

给定一个链表的头指针 ListNode* pHead，请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变。

设置两个链表头，遍历原链表，一个追加小数链表，一个追加大数链表，最后将小数链表粘到大数链表前边即为结果。

public class Partition {
    public ListNode partition(ListNode head, int x) {
        // write code here
        ListNode small = new ListNode(-1);
        ListNode big = new ListNode(-1);
        ListNode smallHead = small;
        ListNode bigHead = big;
        while(head!=null){
            if(head.val<x){
                small.next = head;
                small = small.next;
                head = head.next;
                small.next = null;
            }
            else{
                big.next = head;
                big = big.next;
                head = head.next;
                big.next = null;           
            }
        }
        
        if(bigHead.next==null) return smallHead.next;
        if(smallHead.next==null) return bigHead.next;
        
        
        //连接2个
        ListNode cur = smallHead;
        while(cur.next!=null)
            cur = cur.next;
        cur.next = bigHead.next;
        return smallHead.next;
    }
}