著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的 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 个空格分隔,行首尾不得有多余空格。

输入样例:

  1. 5
  2. 1 3 2 4 5

输出样例:

  1. 3
  2. 1 4 5

思路

  • 从左往右遍历一遍,找每个元素的左边最大值
  • 从右往左遍历一遍,找每个元素的右边最小值
  • 再从左往右遍历一遍,当该项元素的 左边最大值 < 该项 && 右边最小值 > 该项,则把该项加入到 答案数组 里,主元计数器+1
  • 输出答案数组

代码

  1. #include<cstdio>
  2. const int INF = 0x3fffffff;
  3. int main() {
  4. // Input the data
  5. int number;
  6. int input[100001];
  7. scanf("%d", &number);
  8. for(int i = 0; i < number; i++) {
  9. scanf("%d", &input[i]);
  10. }
  11. // Analyze the leftMax and rightMin
  12. int leftMax[100001], rightMin[100001];
  13. leftMax[0] = 0; // There is nothing bigger than input[0]
  14. for(int i = 1; i < number; i++) {
  15. if(leftMax[i - 1] > input[i - 1]) {
  16. leftMax[i] = leftMax[i - 1];
  17. }
  18. else {
  19. leftMax[i] = input[i - 1];
  20. }
  21. }
  22. rightMin[number - 1] = INF; // There is nothing smaller ther input[number - 1]
  23. for(int i = number - 2; i >= 0; i--) {
  24. if(rightMin[i + 1] > input[i + 1]) {
  25. rightMin[i] = input[i + 1];
  26. }
  27. else {
  28. rightMin[i] = rightMin[i + 1];
  29. }
  30. }
  31. int answer[100001] = {0};
  32. int counter = 0; // Count the number of pivot
  33. for(int i = 0; i < number; i++) {
  34. if(leftMax[i] < input[i] && rightMin[i] > input[i]) {
  35. answer[counter] = input[i];
  36. counter++;
  37. }
  38. }
  39. // Display the result
  40. printf("%d\n", counter);
  41. for(int i = 0; i < counter; i++) {
  42. printf("%d", answer[i]);
  43. if(i != counter - 1) {
  44. printf(" ");
  45. }
  46. }
  47. printf("\n");
  48. return 0;
  49. }