PAT乙1045：快速排序

问题描述：

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

例如给定 $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

问题分析：

题目意思很容易理解，就是一个数组中的数和该数在数组中左右的数进行大小比较，如果比左边的都大，比右边的都小，就记录下来，最简单的思路就是套几层循环，挨个个比较，但是这种方法的时间复杂度必然很高，直接不要考虑了。那么得换个思路考虑这个问题，如果说比左边的都大，只要比左边最大的那个大就可以了，右边也是同样的道理。所以解题的思路就自然出来了。

这题的解题思路是，先从左向右进行比较，满足比左边最大的要大，则进行标记；再从右向左进行比较，如果比最右边最小的大，则去除标记。然后统计总数，输出即可。

这题有个坑，我个人认为题目描述不当，即当没有主元时需要输出为cout<<"0"<<endl<<endl；即要输出2个回车。0的问题有考虑到，但是输出2个回车确实有点的过分了。我个人认为，PAT乙级题目中出现的一些坑的问题，有些可以理解为卡分或者是增加考生的区分度，有些明显是出题者的个人失误。考查点过于偏了。

代码：

#include <iostream>
#include <algorithm>
using namespace std;

int main()
{
	int n, input[100010], output[100010], step = 0, max, min;
	bool flag[100010] = { false };//记录是否为主元
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> input[i];
	flag[0] = true;
	max = input[0];//左边最大的
	for (int i = 1; i < n; i++)//从右往左进行比较
		if (input[i] > max)
		{
			flag[i] = true;//满足要求的为true
			max = input[i];
		}
	min = input[n - 1];//右边最小的
	for (int i = n-2; i >= 0; i--)//从右向左进行比较
	{
		if (input[i] < min)
			min = input[i];
		else
			flag[i] = false;//不满足要求的进行排除
	}
	for (int i = 0; i < n; i++)
		if (flag[i])
			output[step++] = input[i];
	cout << step ;
	if (step != 0)
	{
		sort(output, output + step);
		cout << endl << output[0];
	}
	else
		cout << endl << endl;
	for (int i = 1; i < step; i++)
		cout << " " << output[i];
	return 0;
}