著名的快速排序算法里有一個經典的划分過程:我們通常采用某種方法取一個元素作為主元,通過交換,把比主元小的元素放到它的左邊,比主元大的元素放到它的右邊。 給定划分后的 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
//眾所周知,在快速排序中選擇的主元排序后位置是不變的,所以可以根據此來判斷,但是只根據這一種情況來判斷是錯誤的。比如3,2,1,4,5,中2就 //不是主元,所以還需判斷,那個數是不是排序前當前位置最大值 #include<iostream> #include<algorithm> using namespace std; int main(){ int N; cin>>N; int a[N],b[N],c[N];//最好的情況,可能每個數都是主元 for(int i=0;i<N;i++){ cin>>a[i]; b[i]=a[i]; } sort(a,a+N); int count=0; int temp=0,max=0; for(int i=0;i<N;i++){ if(max<b[i]) max=b[i]; //把最大值保存下來 if(a[i]==b[i]&&a[i]==max){//如果和排序后的相同位置一樣,而且是最大值,則確定是主元 c[count++]=a[i]; } } cout<<count<<endl; if(count==0) cout<<endl; //如果主元個數是0,空行還是要輸出的 for(int i=0;i<count;i++){ if(i!=0) cout<<" "; cout<<c[i]; } return 0; }