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;
     }
 }