著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列,请问有多少个元素可能是划分前选取的主元?
例如给定 , 排列是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
思路详解:一般来说一开始并不会马上就想到巧解,暴力求解的思路是,遍历一遍输入,顺便记下左边最大的数,如果正在遍历的数小于最大的,就跳过,否则判断右边是否全部大于正在访问的数,这样做的结果是测试点1运行超时。
所以换方法:将输入升序排序为 1 2 3 4 5 考虑输入 为 1 3 2 4 5 输出为1 4 5,是否只要满足输入状态下的数所在位置与排序好的位置相同即可?于是照此方法写 有四个样例过不了,所以继续思考。
如果输入: 1 3 2 4 7 6 5
排序后是: 1 2 3 4 5 6 7
输出为 1 4 6为主元。但实际上6的左边7比它大,6不应该是主元。可以看到输入状态下的数所在位置与排序好的位置相同仅仅是必要条件,我们还需要判断输入状态下的数的左边是否都比它小(或者右边都比它大),并且不必纠结是否需要两边都判断,正如上面我给出的例子中4的位置确定了,如果左边都小于了4 那么右边必定大于4,因为在排好序的数列中比4小的只有3个,此时输入中的4的位置与排好序的数列中4的位置对应,左边必须需要3个小于4的数,那么大于4的数一定会被放在右边,所以不可能出现4的右边还有比4小的数。
综上所诉,我们可以写出程序了:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
int n;
cin >> n;
int* backup = new int[n];
int* input = new int[n];
vector<int>result;
for (int i = 0; i < n; i++)
{
int a;
cin >> a;
backup[i] = a;
input[i] = a;
}
sort(backup, backup + n);
int max = 0;
for (int i = 0; i < n; i++)
{
if (input[i] > max)
max = input[i];
if (backup[i] == input[i] && input[i] >= max)
result.push_back(backup[i]);
}
cout << result.size() << endl;
for (int i = 0; i < result.size(); i++)
{
if (i == 0)
cout << result[i];
else
cout << " " << result[i];
}
//cout << endl;
delete []input;
delete []backup;
return 0;
}
注意:按照题意,本不需要在程序最后输出换行,但是这样提交的程序测试点2过不了,提示格式错误,需要加上才能AC。说实话挺坑的。。。