链表划分

链表划分

题目：
描述
给定一个单链表和数值x，划分链表使得所有小于x的节点排在大于等于x的节点之前。你应该保留两部分内链表节点原有的相对顺序。

样例
样例 1：
输入：
list = null
x = 0
输出：
null
解释：
空链表本身满足要求

样例 2：
输入：
list = 1->4->3->2->5->2->null
x = 3
输出：
1->2->2->4->3->5->null

/**
 * Definition for ListNode
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         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) {
        // write your code here
        if(head == null) {
            return null;
        }

        ListNode root1 = new ListNode(0), root2 = new ListNode(0);
        ListNode cur = head, cur1 = root1, cur2 = root2;

        while(cur != null) {
            if(cur.val < x) {
                cur1.next = cur;
                cur = cur.next;
                cur1 = cur1.next;
                cur1.next = null;
            } else {
                cur2.next = cur;
                cur = cur.next;
                cur2 = cur2.next;
                cur2.next = null;
            }

        }

        cur1.next = root2.next;
        return root1.next;
    }
}