1045 快速排序 (25 分) (测试点2)

PAT乙级真题

题目链接：

https://pintia.cn/problem-sets/994805260223102976/problems/994805278589960192

引入：

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

例如给定 N=5, 排列是1、3、2、4、5。则：

• 1 的左边没有元素，右边的元素都比它大，所以它可能是主元；
• 尽管 3 的左边元素都比它小，但其右边的 2 比它小，所以它不能是主元；
• 尽管 2 的右边元素都比它大，但其左边的 3 比它大，所以它不能是主元；
• 类似原因，4 和 5 都可能是主元。

因此，有 3 个元素可能是主元。

输入格式：

输入在第 1 行中给出一个正整数 N（≤10
⁵；第 2 行是空格分隔的 N 个不同的正整数，每个数不超过 10⁹。

输出格式：

在第 1 行中输出有可能是主元的元素个数；在第 2 行中按递增顺序输出这些元素，其间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

5
1 3 2 4 5

输出样例：

3
1 4 5

分析：

这道题如果直接遍历比较的话很容易超时，注意题目叫做快速排序。排序就要想到顺序问题，再读一遍题目可以发现其实主元比前面小比后面大就相当于数字排序。
定义一个数组a存储原始数据，定义一个数组b拷贝数据，将b数组进行排序，从0位置开始逐一比较a[i]和b[i]，如果a[i]等于b[i]且a[i]比前面的最大值大，那么a[i]就是主元，同时再更新最大值即可。

测试点2：
测试点2应该是考虑到0个元素的情况输出空行，最后加个输出换行就行

#include <iostream>
#include <algorithm>
using namespace std;
int main(){
	int n, sum = 0, max = 0;
	cin >> n;
	int a[n], b[n], num[n];
	for(int i = 0; i < n; i++){
		scanf("%d", &a[i]);
		b[i] = a[i]; 
	}
	sort(a, a+n);
	for(int i = 0; i < n; i++){
		if(b[i] == a[i] && a[i] > max)
			num[sum++] = b[i];
		if(b[i] > max)
			max = b[i];
	}
	printf("%d\n", sum);
	for(int i = 0; i < sum; i++){
		if(i != 0) printf(" ");
		printf("%d", num[i]);
	}
	if(sum == 0) printf("\n");	
}

悄咪咪表白柳婼大佬