[Leetcode]86. Partition List

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.

 

思路：用指针 P 遍历这个队列，每当值有大于或等于 x 的节点，我们就让 P 停下来，

然后设置一个指针 Q 寻找从 P 以后的第一个值小于 x 的 节点。例如，当 P 停在 4 

时，Q 应当停在 2 ；然后我们设置个指针 swap 从 P 走到 Q，交换 swap 节点和 Q

节点的值。

 // 看leetcode 评论里另一个思路是把小于 x 的所有元素串成一条链，大于等于 x 的

    串成另一条链，最后两链一接，完事。。。。。妙啊！

 

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode partition(ListNode head, int x) {
        if (head == null || head.next == null)
            return head;
        ListNode p = head;
        while (p != null){
            if (p.val >= x){
                ListNode q = p;
                while (q != null){
                    if (q.val < x)
                        break;
                    q = q.next;
                }
                if (q == null)
                    break;
                ListNode swap = p;
                while (swap != q){
                    int tmp = swap.val;
                    swap.val = q.val;
                    q.val = tmp;
                    swap = swap.next;
                }
            }
            p = p.next;
        }
        return head;
    }
}