PAT basic-level 1045 快速排序 笔记

1045. 快速排序(25)

时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CAO, Peng
著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。 给定划分后的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

---

25分题都这么蛋疼的吗？这个题我试了两个方法：
思路1：建立结构体，包括一个int值和一个tag，输入的数组会逐个被存进这个结构体，并且tag初始化为0。
建立两层循环对结构体数组进行遍历，外层i从0到n，内层j从i+1到n，意思是，固定一个元素i，其右元素j的int值如果大于该元素的int值，不做处理，j++，当出现元素j的int值小于元素i的int值，isTag = 1，表示该元素也不是主元元素，建立标记留待后续处理元素i，然后将该j元素的tag标为1，表示为非主元元素。j继续遍历，每出现满足条件的便标记tag = 1，当j遍历到数组结尾，如果isTag = 1，将元素i的tag值标为1，i++。
这里要在遍历时先判断该元素的tag是否为1，如果是 ，continue跳过，这样才能防止运行超时。
但是！！这个方法一直无法通过最后一个测试点，非常迷，一直找不到问题在哪，网上找了找，有个类似的方法做的是正反遍历，但我个人感觉我的做法正着就可以找到所有非主元元素，如果有错误，希望读者评论指正！
思路2：做个拷贝数组，排序后原数组对比，相同的元素记录下来，然后遍历该元素其左边部分，如果该元素是最大值，则是主元元素。
这个方法是网上看来的，一般我只看思路不看代码，我马上写了个，最后一个测试点一下子通过，喜极而泣:(，但第二个和第三个超时了，我修改了下，感觉这个自己这个加速的做法很棒，这里记录下：
建立一个max值，在遍历过程中，每次进行循环先判断该元素是否大于max，若大于，则max=该值，当出现相同值时，只要判断这个值是否==max（因为进循环先对max处理），就知道这个值是否是其左边部分中最大值，这样O(N^2)的复杂度直接降到了O(N)
下面是思路2的AC代码：

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

int main(){
    int n;
    cin >> n;
    vector<int> inp(n, 0);
    for (int i = 0; i < n; i++)
        scanf("%d", &inp[i]);

    vector<int> test(n, 0);
    test = inp;
    sort(test.begin(), test.end());
    vector<int> zhu;
    int max = 0;
    for (int i = 0; i < n; i++){
        if (inp[max] < inp[i])  //保证inp[max]在inp[0-i]最大
            max = i;
        if (inp[i] == test[i] && max == i)
                zhu.push_back(inp[i]);
    }
    if (zhu.size())
        printf("%d\n",zhu.size());
    else {
        printf("%d\n\n", zhu.size());
        return 0;
    }

    sort(zhu.begin(), zhu.end());
    for (vector<int>::const_iterator it = zhu.begin(); it != zhu.end(); it++){
        printf("%d", *it);
        if (it != (zhu.end() - 1))
            printf(" ");
    }

    return 0;
}