题目链接:https://pintia.cn/problem-sets/994805260223102976/problems/994805278589960192
题目描述
著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列,请问有多少个元素可能是划分前选取的主元?
例如给定 N = 5 N = 5 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
代码
#include <stdio.h>
#include <algorithm>
using namespace std;
int const maxn = 100001;
int a[maxn], left[maxn], right[maxn], temp[maxn];
int main() {
int n;
scanf("%d", &n);
for(int i = 0; i < n; i++)
scanf("%d", &a[i]);
int min = 1000000000, max = 0, count = 0; //min初始要足够大,否则会答案错误
for(int i = n - 2; i >= 0; i--){
if(a[i + 1] < min)
min = a[i + 1];
right[i] = min;
}
if(a[0] < right[0])
temp[count++] = a[0];
for(int i = 1; i < n; i++){
if(a[i - 1] > max)
max = a[i - 1];
left[i] = max;
if(a[i] > left[i] && a[i] < right[i])
temp[count++] = a[i];
}
if(a[n - 1] > left[n - 1])
temp[count++] = a[n - 1];
sort(temp, temp + count);
printf("%d\n", count);
for(int i = 0; i < count; i++) {
if(i > 0)
printf(" ");
printf("%d", temp[i]);
}
printf("\n"); //最后要换行,否则答案错误
return 0;
}
【注】:OJ系统提示“段错误”,可能因为某个函数内开的数组过大,导致该函数的栈无法容纳数组,造成爆栈。
解决方法:把数组开到函数外边,此时数组保存在全局变量区。