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

题目描述：

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

我的思路：

• 如果是主元，排序前后位置不变（前边的元素都比它小，后边的元素都比它大，前后元素个数也相同）。
• 如果输入的元素个数为0，则应在第二行输出换行。
• 详细思路见代码

我的代码：

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

int main(){
	int n;
	cin>>n;
	int vs[n],sorted[n],answer[n];
	for(int i=0;i<n;i++){
		cin>>vs[i];//vs存放原数组
		sorted[i]=vs[i];
	}
	sort(sorted,sorted+n);//sorted存放排序后的数组
	int maxn=-1,cnt=0;//maxn存放该元素之前的最大值，cnt记录解的个数
	for(int i=0;i<n;i++){
		if(maxn<vs[i]){//更新maxn的值
			maxn=vs[i];
		}
		//最大值与该元素值相同，且排序前后位置不变——满足条件
		if(maxn==vs[i]&&sorted[i]==vs[i]){
			answer[cnt++]=vs[i];
		}
	}
	cout<<cnt<<endl;
	for(int i=0;i<cnt;i++){
		if(i==0){
			cout<<answer[i];
		}else{
			cout<<" "<<answer[i];
		}
	}
	if(cnt==0){//若主元个数为0，则需要第二行需要输出换行
        cout<<endl;
    }
	return 0;
}