PAT B1045

1045. 快速排序(25)

著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。给定划分后的N个互不相同的正整数的排列，请问有多少个元素可能是划分前选取的主元？

例如给定N = 5, 排列是1、3、2、4、5。则：

• 1的左边没有元素，右边的元素都比它大，所以它可能是主元；
  
• 尽管3的左边元素都比它小，但是它右边的2它小，所以它不能是主元；
  
• 尽管2的右边元素都比它大，但其左边的3比它大，所以它不能是主元；
  
• 类似原因，4和5都可能是主元。

  因此，有3个元素可能是主元。

  输入格式：

  输入在第1行中给出一个正整数N（<= 10⁵）；第2行是空格分隔的N个不同的正整数，每个数不超过10⁹。

  输出格式：

  在第1行中输出有可能是主元的元素个数；在第2行中按递增顺序输出这些元素，其间以1个空格分隔，行末不得有多余空格。

  输入样例：

  5
  1 3 2 4 5
  

  输出样例：

  3
  1 4 5
  

---

代码如下：

#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN = 100010;
const int INF = 0x3fffffff;
int a[MAXN], leftMax[MAXN], rightMin[MAXN];
int ans[MAXN], num = 0;
int main(){
  int n;
  scanf("%d", &n);
  for(int i = 0; i < n; i++){
    scanf("%d", &a[i]);
  }
  leftMax[0] = 0;
  for(int i = 1; i < n; i++){
    leftMax[i] = max(leftMax[i - 1], a[i - 1]);
  }
  rightMin[n - 1] = INF;
  for(int i = n - 2; i >= 0; i--){
    rightMin[i] = min(rightMin[i + 1], a[i + 1]);
  }
  for(int i = 0; i < n; i++){
    if(leftMax[i] < a[i] && rightMin[i] > a[i]){
      ans[num++] = a[i];
    }
  }
  printf("%d\n", num);
  for(int i = 0; i < num; i++){
    printf("%d", ans[i]);
    if(i < num - 1) printf(" ");
  }
  printf("\n");
  return 0;
}