著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列,请问有多少个元素可能是划分前选取的主元?
例如给定 $N = 5$, 排列是1、3、2、4、5。则:
- 1 的左边没有元素,右边的元素都比它大,所以它可能是主元;
- 尽管 3 的左边元素都比它小,但其右边的 2 比它小,所以它不能是主元;
- 尽管 2 的右边元素都比它大,但其左边的 3 比它大,所以它不能是主元;
- 类似原因,4 和 5 都可能是主元。
因此,有 3 个元素可能是主元。
输入格式:
输入在第 1 行中给出一个正整数 N(≤1e+5); 第 2 行是空格分隔的 N 个不同的正整数,每个数不超过 109。
输出格式:
在第 1 行中输出有可能是主元的元素个数;在第 2 行中按递增顺序输出这些元素,其间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:
5
1 3 2 4 5
输出样例:
3
1 4 5
//题意:找到一组数中满足下列条件的数:左边元素都比它小 且 右边元素都比它大,然后顺序输出
//思路:对于数组中每一个数,与其左边最大数和右边最小数比较
//实现:1、暴力解法:对于数组中每一个元素,遍历它之前的所有元素,找到最大的,然后遍历它之后的元素,找到最小的,这 样的时间复杂度是O(n * n),对于最大数据1e+5,肯定会TLE(超时)。因此此方法并不能得到满分 2、我们可以这么做,在读入数组中的元素的同时,找到此数之前(包括这个数)的最大值,这个实现并不难,代码中 有详细介绍(这里先不说了)。同理,逆序遍历数组,找到每个元素右边的最小数。时间复杂度是O(n)
//下面是我AC的代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 100005 //数组最大容量
int ans[MAX], a[MAX], lmax[MAX]; //lmax[i]表示下标为i的元素左边(包括它自身)的最大值,ans储存结果
int main() {
int n, k = 0; //k为已找到满足条件的数的个数,初始化为0
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i]; //读入初始数组中的元素
if (!i)
lmax[0] = a[0]; //lmax[0]的值就是a[0], 关键
else
lmax[i] = max(lmax[i - 1], a[i]); //lmax[i]是lmax[i - 1]和新读入的数据(即a[i])的较大者
}
int rmin = a[n - 1]; //rmin是当前右边的最小值,不必用数组表示,因为是逆序遍历数组,每次都能直接更新
for (int i = n - 1; i >= 0; i--) {
if (lmax[i] == a[i]) { //这个条件成立则说明下标为i的这个元素左边的所有数都比它小(注意题目中说明的数组元素不同)
if(i == n - 1) //如果是最后一个元素的话,就满足题意
ans[k++] = a[i];
else if (rmin > a[i]) //不是最后一个数的话,就要判断右边的最小值和它的大小关系
ans[k++] = a[i];
}
rmin = min(a[i], rmin); //更新最小值
}
if (k) { //如果有满足题意的数
sort(ans, ans + k); //先顺序排序
cout << k << endl << ans[0];
for (int i = 1; i < k; i++)
cout << " " << ans[i]; //输出ans中的元素,注意最后一个元素后无空格
cout << "\n";
}
else //没有满足的(其实就是初始数组是降序的),就输出0
cout << 0 << "\n\n"; //注意要输出一个空行
return 0;
}
//先看后赞是好习惯哦!!