PAT 1045 快速排序 (25分)（Java）

題目描述：

  著名的快速排序算法裏有一個經典的劃分過程：我們通常採用某種方法取一個元素作爲主元，通過交換，把比主元小的元素放到它的左邊，比主元大的元素放到它的右邊。 給定劃分後的 N 個互不相同的正整數的排列，請問有多少個元素可能是劃分前選取的主元？
  例如給定 $N = 5$, 排列是1、3、2、4、5。則：
  1 的左邊沒有元素，右邊的元素都比它大，所以它可能是主元；
儘管 3 的左邊元素都比它小，但其右邊的 2 比它小，所以它不能是主元；
儘管 2 的右邊元素都比它大，但其左邊的 3 比它大，所以它不能是主元；
類似原因，4 和 5 都可能是主元。
  因此，有 3 個元素可能是主元。

輸入格式：

  輸入在第 1 行中給出一個正整數 N（≤10^5）； 第 2 行是空格分隔的 N 個不同的正整數，每個數不超過 10^9。

輸出格式：

  在第 1 行中輸出有可能是主元的元素個數；在第 2 行中按遞增順序輸出這些元素，其間以 1 個空格分隔，行首尾不得有多餘空格。

輸入樣例：

5
1 3 2 4 5

輸出樣例：

3
1 4 5

具體實現：

package com.hbut.pat;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Pat_1045 {
    public static void main(String[] args) throws Exception {
        int max = 0, index = 0;
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[] a = new int[n], b = new int[n], v = new int[n];
        String[] in = br.readLine().split(" ");
        for (int i = 0; i < n; i++) {
            a[i] = Integer.parseInt(in[i]);
            b[i] = a[i];
        }
        Arrays.sort(a);
        for (int i = 0; i < n; i++) {
            if (a[i] == b[i] && b[i] > max)
                v[index++] = b[i];
            if (b[i] > max)
                max = b[i];
        }
        System.out.println(index);
        for (int i = 0; i < index - 1; i++) {
            System.out.print(v[i] + " ");
        }
        System.out.println(v[index - 1]);
    }
}