LintCode-链表划分

容易 链表划分 查看运行结果

29% 通过
给定一个单链表和数值x，划分链表使得所有小于x的节点排在大于等于x的节点之前。

你应该保留两部分内链表节点原有的相对顺序。

您在真实的面试中是否遇到过这个题？ Yes
样例
给定链表 1->4->3->2->5->2->null，并且 x=3

返回 1->2->2->4->3->5->null

这里对题目理解有点问题：

案例一：
输入
1->4->3->2->5->2->null, 3
输出
1->4->2->2->3->5->null
期望答案
1->2->2->4->3->5->null


案列二
输入
1->2->0->3->1->2->1->0->2->2->2->1->0->2->null, 2
输出
0->0->0->1->1->1->1->2->3->2->2->2->2->2->null
期望答案
1->0->1->1->0->1->0->2->3->2->2->2->2->2->null


不理解他划分好到底需不需要排序，能正确理解的朋友请留言指教。

/**
 * Definition for ListNode.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int val) {
 *         this.val = val;
 *         this.next = null;
 *     }
 * }
 */ 
public class Solution {
    /**
     * @param head: The first node of linked list.
     * @param x: an integer
     * @return: a ListNode 
     */
    public ListNode partition(ListNode head, int x) {
         if(head == null){
             return null;
         }
         ArrayList<Integer> a = new ArrayList<Integer>();
         ArrayList<Integer> b = new ArrayList<Integer>();
         boolean find =false;
        while(head != null){    
            if(head.val == x && !find){
                find =true;
                head = head.next;
                continue;
            }else{
                if(find){
                    if(head.val < x){
                        a.add(head.val);
                    }else{
                        b.add(head.val);
                    }
                }else{
                    a.add(head.val);
                }
                head = head.next;
            }
        }
        ListNode re = new ListNode(0);
        ListNode ree = re;
        for(int n : a){
            re.next = new ListNode(n);
            re = re.next;
        }
        insertionSortList(ree);
        if(find){
            re.next = new ListNode(x);
            re = re.next;
        }
        for(int n : b){
            re.next = new ListNode(n);
            re = re.next;
        }
        return ree.next;
     }

    public static ListNode insertionSortList(ListNode head) {
        boolean flag = true;
        ListNode frist = head;
        ListNode second = head;
        if(head == null || head.next == null){
            return head;
        }else{
            frist = head.next;
        }
        while(flag){
            flag = false;
            do{
                if(frist.val<second.val){
                    int x =frist.val;
                    frist.val = second.val;
                    second.val = x;
                    flag = true;
                }           
                frist = frist.next;
                second = second.next;
            }while(frist != null);
            frist = head.next;
            second = head;
        }
        return head;
    }
}