PAT 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

题目分析：
这道题目如果暴力解题，毫无疑问时间复杂度会很高，那么我们就要想办法来降低程序运行的次数。
熟悉的快速排序的读者应该知道，快速排序每选择一次的主元进行一轮排序之后，主元的位置就是最终的位置。那么我们可以利用这个特点排除掉一大部分的无用计算。即直接使用sort得到最终的序列，然后核对前后的位置有没有发生变化，变化的一定不可能是主元。但是位置没有变化的也有一些不时主元，比如序列 1 4 3 2 5 中的3虽然位置没有变化，但是前面有比它大的4，所以它也不可能是主元。所以我们还需要多加一部判断。笔者是一开始就设置的结构体，成员位：记录初始位置的location，记录数据的num，以及标记初始序列的该元素是不是都比它之前元素大的flag。
另外注意的是：测试点2的格式，很容易就能实验猜到测试点2的数据主元个数为零，但是输出主元的位置仍然要流出来，就是在最后在加一句printf（“\n”）；
请看下面代码：

#include<stdio.h>
#include<algorithm>
using namespace std;
const int maxn=100001;
struct number{             //结构体，成员存放数据的num，初始序列中的位置location，标记该元素是否都比之前元素要大的flag 
    int num,location;
    bool flag;
}data[maxn];
bool cmp(number a,number b){   //比较函数 
    return a.num <b.num ;
}
int main()
{
    int n,flag=0,count=0,max=0; //个数n，控制输出空格的flag，主元个数count，当前最大值max 
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&data[i].num);    //读入一个整数 
        if(max<=data[i].num){        //更新max 
            max=data[i].num;
            data[i].flag=true;        //该整数比之前的元素都要大，flag=true 
        }
        data[i].location=i;          //记录初始位置 
    }
    sort(data,data+n,cmp);           //排序 
    for(int i=0;i<n;i++){
        if(data[i].location==i&&data[i].flag){  //位置不变，且有比之前的元素都要大，则是主元 
            count++;
        }
    }       
    printf("%d\n",count);        //输出主元个数 
    for(int i=0;i<n;i++){
        if(data[i].location==i && data[i].flag ){
            if(flag==0){ 
                printf("%d",data[i].num );   //第一个主元，之前不输出空格 
                flag=1;
            } 
            else printf(" %d",data[i].num );  //之后的每一个主元前都要输出空格 
        }
    }
    printf("\n");           //不加这一句，测试点2过不去 
}