1045. 快速排序(25)
著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的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
题解:
先讲讲我开始的思路:我开始的思路是从第一个数字开始判断其左右的数字,但是后来发现在测试点会有运行超时,所以我果断放弃了,
因为五位数字的循环中还嵌套着大循环确实耗时巨大。
后来我开始想其他办法。写完后,在百度,发现跟网上大部分ac的人是一样的。主要思路就是利用排序完的数组和排序前的数组进行对比,
随着i的递增,如果两个数组的值一样并且检查过的数字的最大值与排序前的数组的a[i]相同,这些值就是主元。
代码:
#include <iostream>
#include <algorithm>
using namespace std;
int main(){
int n;
int max=-1;
int a[100001],b[100001],results[100001];
int r=0;
cin>>n;
for(int i=0;i<n;i++){
cin>>a[i];
b[i]=a[i];
}
sort(b,b+n);
for(int i=0;i<n;i++){
if(max<a[i])max=a[i];
if(a[i]==b[i]&&max==a[i])results[r++]=a[i];//关键语句:利用排序后的数组与排序前的数组比较,得出过程中的主元。
}
cout<<r<<endl;
for(int i=0;i<r;i++){
if(i==0)
cout<<results[i];
else
cout<<" "<<results[i];
}
cout<<endl; //如果不写,会出现格式错误。(测试点2)
return 0;
}