PAT乙级 1045 快速排序 (25分)

1045 快速排序 (25分)

著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列，请问有多少个元素可能是划分前选取的主元？

例如给定 $N=5$, 排列是1、3、2、4、5。则：

1 的左边没有元素，右边的元素都比它大，所以它可能是主元；
尽管 3 的左边元素都比它小，但其右边的 2 比它小，所以它不能是主元；
尽管 2 的右边元素都比它大，但其左边的 3 比它大，所以它不能是主元；
类似原因，4 和 5 都可能是主元。
因此，有 3 个元素可能是主元。

输入格式：

输入在第 1 行中给出一个正整数 N（≤10​⁵​​ ）； 第 2 行是空格分隔的 N 个不同的正整数，每个数不超过 10​⁹​​ 。

输出格式：

在第 1 行中输出有可能是主元的元素个数；在第 2 行中按递增顺序输出这些元素，其间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

5
1 3 2 4 5

输出样例：

3
1 4 5

思路：

先说一下坑点：如果主元数量为0的话，输出0以后还要多输出一个空行。
我至少找到了三种解决办法：

1. 第一种就是对每个位置的数寻找左边最大和右边最小的数，通过维护两个数组，一个是当前元素左边最大的数的数组，一个是当前元素右边最小的数的数组。时间复杂度是O(n)
2. 第二种方法是先排序，对每个排好序数组和原数组比较，如果同样的位置上的数相同，并且比所有左边是数据大，那么就是主元。时间复杂度是O(nlnn)
3. 第三种方法是不断维护主元数组，把数据一个一个地输入，记录最大值max。读取下一个数，若下一个数比max大就放进主元数组中，若下一个数比max小，就把主元数组中，所有比这个数小的主元踢出去。时间复杂度也是O(n)，而且空间用的应该是最少的。
   因为数据量比较小的原因(或者其他原因)，这三种方法耗时都是差不多的。

统计数据：

题解 方法一：

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

int n,num[101000],mi[101000],ma,out[101000],cnt;

int main(){
    scanf("%d",&n);
    for(int i=0;i<n;++i)
        scanf("%d",&num[i]);
    mi[n] = 1e9+7;
    for(int i=n-1;i>=0;--i)      //mi数组的意思是，从当前位置一直到最右边中，最小的数。
        mi[i] = num[i] < mi[i+1] ? num[i] : mi[i+1];
    for(int i=0;i<n;++i){
        if(num[i]>ma && num[i]<mi[i+1])
            out[cnt++] = num[i];
        ma = num[i] > ma ? num[i] : ma;
    }                      //你可能发现我没有对out数组排序，因为out数组一定是单调递增
    printf(cnt?"%d\n":"%d\n\n",cnt); //cnt是0的时候要多输出一个换行
    for(int i=0;i<cnt;++i)
        printf("%s%d",i?" ":"",out[i]);
    return 0;
}

题解 方法二：

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

int n,num[101000],so[101000],out[101000],cnt,ma;

int main(){
    scanf("%d",&n);
    for(int i=0;i<n;++i){
        scanf("%d",&num[i]);
        so[i] = num[i];
    }
    sort(so,so+n);
    for(int i=0;i<n;++i){
        if(so[i]==num[i] && so[i]>ma)  //跟原来位置相同，并且小于ma的一定是主元
            out[cnt++] = so[i];
        ma = max(num[i],ma);
    }
    printf(cnt?"%d\n":"%d\n\n",cnt); //cnt是0的时候要多输出一个换行
        printf("%s%d",i?" ":"",out[i]);
    if(cnt==0) printf("\n");
    return 0;
}

题解 方法三：

#include<algorithm>
#include<cstdio>
using namespace std;
#define maxn 100005

int n,maxx,cnt,out[maxn],tmp;

int main(){
    scanf("%d",&n);
    for(int i = 0;i<n;i++){
        scanf("%d",&tmp);
        if(tmp>maxx){            //小于maxx的都放进主元数组
            out[cnt++] = tmp;
            maxx = tmp;
        }
        for(int j = cnt-1;j>=0;j--)   //把主元数组中，小于tmp的都踢掉
            if(out[j]>tmp)
                cnt--;
            else
                break;
    }                         //out数组也没必要排序，因为一定是从小到大排好了的
    printf(cnt?"%d\n":"%d\n\n",cnt);
    for(int i = 0;i<cnt;i++)
        printf("%s%d",i?" ":"",out[i]);
}