PAT 1045 快速排序（测试点2 3 5 已过）C++

补前未做+测试点2/3/5
——————
题目连接：请点击
思路：由题目描述可知，主元可能是那些即使排好序之后位置仍不变的（它原本就呆在此位置上），这样就可以做到前面的数都比它小，后面的都比它大。但这样想有个例外（这也是起初2/3/5测试点未过的原因之一），若给出的数列是3,2,1则排序后1,2,3，发现如果只对比位置是否变化，则把2当成主元（其实2不是），因此也要比较此位置的数是否大于此前元素的最大值。
注解：
1 测试点2还有个坑就是格式问题。即使无主元，同样要输出可能主元个数0以及下一行无元素（占1行）。cout<<count<<endl<<endl;
2 起初未比较特定位置与此前元素的最大值时，测试点2/3/5未过，所以，猜想测试点2/3/5应该具有刚好是从大到小的数列且该数列个数恰好为奇数（比如3,2,1）

#include<iostream>
#include<algorithm>
using namespace std;
int main(){
	int N,count=0,max=0;//count表示符合题要求主元个数 
	cin>>N;
	int num[N],sortNum[N];
	for(int i=0;i<N;i++){
		cin>>num[i];
		sortNum[i]=num[i];	
	} 
	sort(sortNum,sortNum+N);
	int priKey[N];//priKey用于存放主元
	if(sortNum[0]==num[0]) priKey[count++]=num[0];
	max=num[0];
	for(int i=1;i<N;i++){
		if(num[i]>max) max=num[i];//找当前数列中的最大值 
		int cnt=i;
		while(cnt<N&&sortNum[cnt]==sortNum[cnt+1]) cnt++;//跳过那些具有相同值的 
		if((sortNum[cnt]==num[cnt])&&(num[i]>=max)) {//若是主元则且本来就应在排好序数列的此位置上且它前面没有比它更大的数 
			for(int j=0;j<(cnt-i)+1;j++) priKey[count++]=num[i];//可能同时有多个相同值的 
		}
		i=cnt;
	}
	if(count==0) cout<<count<<endl<<endl;
	else{
		cout<<count<<endl;
		for(int j=0;j<count;j++){
			cout<<priKey[j];
			if(j!=count-1) cout<<" ";
		}
	}
	return 0;
}