PAT乙级 1045快速排序 测试点2/3/5超时解决

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

本题在下解法为，将输入的N个数据进行排列，排列前后位置索引不变的元素便是主元，但遗漏了一种情况。如数组1 5 3 4 2，排列后是1 2 3 4 5，显然按照索引位置不变即主元判断的话，会得到主元为1 3 4的错误答案，正确答案应为1。也就是，排序前后改变位置的一定不是主元，但主元不一定改变位置。
测试点3和5便是测试的以上问题，主元不仅要排列前后索引位置不变，而且要大于它索引之前的所有数，题目说明了输入的是N个不同的数，不需要考虑元素相同情况
测试点2是格式错误，当主元为0时，需要换两行，也就是两个\n，PAT常规操作(ˉ▽ˉ；)…

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <vector>

using namespace std;

int main() {
    long n;
    cin >> n;
    vector<long> ve1;
    for(long i = 1; i <= n; i++) {
        long num;
        scanf("%ld", &num);//使用cin的话可能会超时
        ve1.push_back(num);
    }
    vector<long> ve2 = ve1;
    sort(ve1.begin(), ve1.end());
    vector<long> result;
    long themax = 0;//记录排列前容器出现的最大值
    //ve2为排列前，ve1为排列后
    for(long i = 1; i <= n; i++) {
    	//主元不仅要排列前后索引不变，也要在原容器中大于它之前的所有元素，解决测试点3、5
        if(ve2.at(i - 1) == ve1.at(i - 1) && ve2.at(i - 1) > themax) {
            result.push_back(ve2.at(i - 1));
        }
        if(ve2.at(i - 1) > themax) {
            themax = ve2.at(i - 1);
        }
    }
    long len = result.size();
    printf("%ld\n", len);
    for(vector<long>::iterator it = result.begin(); it != result.end(); it++) {
        printf("%ld", *it);
        if(it != (--result.end())) {
            printf(" ");
        }
    }
    printf("\n");//最后输出换行，解决测试点2
    return 0;
}

在下辣鸡大一学生，目前正在学数据结构，不懂快速排序算法，做这题才知道原来有种排序叫快速排序(￣▽￣)"

以上有不正确或合适的地方请包涵😘