將單鏈表進行分區

原題

　　Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.
　　You should preserve the original relative order of the nodes in each of the two partitions.
　　For example,
　　Given 1->4->3->2->5->2 and x = 3,
　　return 1->2->2->4->3->5.

題目大意

　　給定一個單鏈表和一個值x，將鏈表分紅小於等於x的部分和大於x的部分。同時保持鏈表原來的相對順序。

解題思路

　　建立兩個鏈表a，b，將原來鏈表中的每一個結點，小於等於x的結點放在a鏈表的末尾，若是是大於就放在b的末尾，最後將b的頭結點接到a末尾。

代碼實現

結點類

public class ListNode {
    int val;
    ListNode next;
    ListNode(int x) { val = x; }
}

• 1
• 2
• 3
• 4
• 5

• 1
• 2
• 3
• 4
• 5

算法實現類

public class Solution {
    public ListNode partition(ListNode head, int x) {

        ListNode le = new ListNode(0); // 小於x的鏈表
        ListNode ge = new ListNode(0); // 大於等於x的鏈表
        ListNode t1 = le;
        ListNode t2 = ge;
        ListNode p = head;

        while (p != null) {
            if (p.val < x) {
                t1.next = p;
                t1 = p;
            } else {
                t2.next = p;
                t2 = p;
            }
            p = p.next;
        }

        t2.next = null;

        // 說明小於的鏈表上有數據
        if (t1 != le) {
            t1.next = ge.next;
            head = le.next;
        } else {
            head = ge.next;
        }
        return head;
    }

}

將單鏈表進行分區