1. 程式人生 > 實用技巧 >1045 快速排序 (25分)

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

解析

  這題用到了當時《有多少PAT》的想法,即將每一位左邊最大數,右邊最小數分別儲存下來,然後遍歷一遍原陣列,如果這個數比左邊最大要大,比右邊最小要小那麼這個數就可能是主元
  將每個可能的主元儲存re陣列,sort一下即可
  注意儲存極值的時候的邊界情況,即第一個數左邊的最大數,最後一個數右邊的最小數,應取為本身。其他情況最大值比較它上一個元素的值a[i - 1]與它上一個元素左邊的最大值l[i - 1]比較,l[i] = max(a[i-1],l[i-1]),同理右邊最小值類似 r[i] = min(a[i+1],r[i+1])

答案

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

    int a[100001];
    int l[100001],r[100001],re[100001];
    int main(){
        int n,k=0;
        cin >> n;
        for(int i = 0 ; i < n ; i ++){
            scanf("%d",a+i);
            if(i){
                l[i] = max(a[i-1],l[i-1]);
            }else{
                l[i] = a[i];
            }
        }//左邊最大值
        for(int i = n - 1 ; i >= 0 ; i --){
            if(i == n-1){
                r[i] = a[i];
            }else{
                r[i] = min(a[i+1],r[i+1]);
            }
        }右邊最小值
        for(int i = 0 ; i < n ; i ++){
            if(a[i] >= l[i] && a[i] <= r[i]){
                re[k++] = a[i];
            }
        }
        cout << k << endl;
    
        sort(re,re+k);

        for(int i = 0 ; i < k; i ++){
            if(i){
                cout<<" "<<re[i];
            }else{
                cout<<re[i];
            }
        }
        cout<<endl;
}