1045 快速排序 (25)——PAT(乙级)

原题链接

题目

著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。 给定划分后的 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

解题思路

这道题是要求找出符合条件的所有正整数 r ：

1. r 左侧数字均小于 r ；
2. r 右侧数字均大于 r .

用一个数组 f 存储 r ，m 记录符合条件的整数个数，mx 记录已输入的数据的最大值。依次输入当前数据 t 。
a. 如果已有满足条件1的数据存放于 f 中，则判断 t 是否比 f 中下标为0到m-1的数据大（如果 f 中数据满足条件1，则其一定是从小到大存储的，则此时只需要判断f[m-1]是否比 t 小，如果不是，则去掉f[m-1]，即m–，一直到m==0或f[m-1]<t为止）。
b. 如果 t 比当前 mx 大，即 t 满足条件1，则 f[m]=t 且 m 加1，同时使 mx = t 。
 
【注】：如果没有满足条件的正整数，即 m=0，此时在第二行需要输出一个换行，否则第三个测试点会显示格式错误。

代码（C++）

#include<iostream>
using namespace std;
int f[100010];
int main() {
	int n,m=0,t,mx=-1;
	cin>>n;
	for(int i=0; i<n; i++) {
		cin>>t;
		while(m>0) {//判断f中数据是否比其后面的数据小
			if(t<f[m-1])	m--;
			else	break;
		}
		if(t>mx) {//判断t是否比左侧数据都大
			mx=t;
			f[m++]=t;
		}
	}
	cout<<m<<endl;
	for(int i=0; i<m; i++) {
		cout<<f[i];
		if(i!=m-1)cout<<" ";
	}
	cout<<endl; //m=0时第二行必须输出换行
	return 0;
}