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

 

        本题考察一些数据结构跟编程技巧，感觉难度不是很大，但通过率比较低也是很奇怪。本题要找的数需要满足在它之前的数都比它小，在它之后的数都比它大。也就是说对于每个数，需要找出它前面的数字序列中的最大值和它后面的数字序列中的最小值。

        我的做法就是定义两个数组num1和num2，num1[i]保存原始序列中第一个数到第i个数之间的最大值，num2[i]保存原始序列中倒数第一个数到第i个数中的最小值（方向相反）。然后判断原始序列中第i个数与num1[i]、num2[i]的大小关系即可。

        第一遍交上去通过之后对我的程序做了一些优化。首先是找最大值的操作可以与输入操作一起完成，因为二者都是在对数组进行顺序的遍历，所以可以放到一个for语句中完成。

        其次是可以省去num2数组，原本我的程序中计算num2使用了一个单独的for循环，之后又用了一个for循环来判断每个数是否符合题意。这个改动是比较大的，下面结合代码来理解一下：

优化前：

for(int i=n-1;i>=0;i--)
{
	min=min<num[i]?min:num[i];
	num2[i]=min;
}
vector<int> res;
for(int i=0;i<n;i++)
{
	if(num[i]>=num1[i] && num[i]<=num2[i])
	res.push_back(num[i]);
}

优化后省去了num2数组，且将两个过程放到一起进行：

vector<int> res;
for(int i=n-1,min=num[n-1];i>=0;i--)
{
	min=min<num[i]?min:num[i];
	if(num[i]>=num1[i] && num[i]<=min)
		res.push_back(num[i]);
}

源代码如下：

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n;
	cin>>n;
	int num[n],num1[n];
	int max=0;
	for(int i=0;i<n;i++)
	{
		scanf("%d",&num[i]);
		max=max>num[i]?max:num[i];
		num1[i]=max;
	}
	vector<int> res;
	for(int i=n-1,min=num[n-1];i>=0;i--)
	{
		min=min<num[i]?min:num[i];
		if(num[i]>=num1[i] && num[i]<=min)
			res.push_back(num[i]);
	}
	sort(res.begin(),res.end());
	cout<<res.size()<<endl;
	for(int i=0;i<res.size();i++)
	{
		cout<<res[i];
		if(i!=res.size()-1)
			cout<<" ";
	}
	cout<<endl;//不加这句会格式错误，但我也不知道为什么要加这句
	return 0;
}