【个人代码及思路】PAT乙级：1045 快速排序 (25分)

PAT (Basic Level) Practice

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

个人代码:

#include<cstdio>

const int MAXN = 100010;

struct ChoosePivot {
    int num;
    int Lflag;  //是否满足num左边都比它小
    int Rflag;  //是否满足num右边都比它大
    int flag;   //是否同时满足条件
} Pivot[MAXN];

int main() {
    int n, cntPivot = 0;
    int Lmax = 0, Rmin = 0; //记录当前数字左边的最大值和右边的最小值
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {  //输入并记录左边的Lflag
        scanf("%d", &Pivot[i].num);
        if(Pivot[i].num > Lmax) {
            Pivot[i].Lflag = 1;
            Lmax = Pivot[i].num;
        } else {
            Pivot[i].Lflag = 0;
        }
    }

    Pivot[n-1].Rflag = 1;
    Pivot[n-1].flag = Pivot[n-1].Lflag;
    Rmin = Pivot[n-1].num;
    if(Pivot[n-1].flag) cntPivot++;

    for(int i = n-2; i >= 0; i--) { //倒着遍历一遍num，记录Rflag和flag
        if(Pivot[i].num < Rmin) {
            Pivot[i].Rflag = 1;
            Rmin = Pivot[i].num;
        } else {
            Pivot[i].Rflag = 0;
        }
        Pivot[i].flag = Pivot[i].Lflag * Pivot[i].Rflag;
        if (Pivot[i].flag == 1) {
            cntPivot++;
        }     
    }

    int temp = cntPivot;
    printf("%d\n", cntPivot);
    if(cntPivot == 0) printf("\n");
    for(int i = 0; i < n; i++) {
        if(Pivot[i].flag == 1) {
            printf("%d", Pivot[i]);
            temp--;
        } else continue;
        if(temp) {
            printf(" ");
        }
    }

    return 0;
}

思路提示:

测试点2，注意0的时候要再换行一次。