著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的N个互不相同的正整数的排列,请问有多少个元素可能是划分前选取的主元?
例如给定N = 5, 排列是1、3、2、4、5。则:
- 1的左边没有元素,右边的元素都比它大,所以它可能是主元;
- 尽管3的左边元素都比它小,但是它右边的2它小,所以它不能是主元;
- 尽管2的右边元素都比它大,但其左边的3比它大,所以它不能是主元;
- 类似原因,4和5都可能是主元。
因此,有3个元素可能是主元。
输入格式:
输入在第1行中给出一个正整数N(<= 105); 第2行是空格分隔的N个不同的正整数,每个数不超过109。
输出格式:
在第1行中输出有可能是主元的元素个数;在第2行中按递增顺序输出这些元素,其间以1个空格分隔,行末不得有多余空格。
输入样例:5 1 3 2 4 5
输出样例:3 1 4 5
思路:快速排序后我元素的位置有一个特点,如果是主元素,那么进行一次排序操作后,主元素的位置和排序完成后,元素的位置不会发生改变。因为他的左边的数都比他大,右边的数都比他小。故:对原序列进行排序后,如果当前元素的位置没有发生改变,并且原序列当前位置左边的所有元素中的最大值比当前元素小,当前元素就是主元素。
#include<cstdio> #include<algorithm> using namespace std; int main(){ int n; scanf("%d",&n); int a[n]; int b[n]; int ans[n];//记录主元素 int cnt = 0; int max = 0; 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(a[i]==b[i]&&b[i]>max){ ans[cnt++] = b[i]; } if(b[i]>max){ max = b[i]; } } printf("%d\n",cnt); for(int i=0 ;i<cnt ;i++){ if(i==0){ printf("%d",ans[i]); }else{ printf(" %d",ans[i]); } } printf("\n"); return 0; }