1045. 快速排序(25)

这道题对时间卡的很严，貌似复杂度大于O(n)会超时。

我的方法是设置一个bool数组对应录入的每个值，从左到右遍历，若当前元素小于左边已出现的最大值，或大于右边已出现的最小值，则他可能是主元，记录这个位置flag=true，否则一定不是主元。遍历一次后再顺序输出即可。

网上还有其他方法，比如排完序后主元位置不变，与当前序列比较，不变的是主元，这些方法也可以参考。

要注意的一个坑是此时主元数量可能为0，所以最后输出时必须判断，若主元为0则第二行输出空行。否则会有一个点格式错误。

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（<= 10⁵）； 第2行是空格分隔的N个不同的正整数，每个数不超过10⁹。

  输出格式：

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

  输入样例：

  5
  1 3 2 4 5
  

  输出样例：

  3
  1 4 5

#include<iostream>
#include<algorithm>
#include<iomanip>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<math.h>
#include<string>
#include<sstream>

using namespace std;

int main(){
    int n;
    cin>>n;
    int *arr=new int[n];
    memset(arr,0,sizeof(arr));
    vector<int> ans;
    for(int i=0;i<n;i++)
    {
        cin>>arr[i];
    }
    int temp=0;
    bool *flag=new bool[n];
    for(int i=0;i<n;i++)
        flag[i]=true;
    int left=arr[0];//左边开始的最大值
    int right=arr[n-1];//右边开始的最小值
    for(int i=0;i<n;i++)
    {
        if(left>arr[i])flag[i]=false;
        else left=arr[i];
        if(right<arr[n-1-i])flag[n-1-i]=false;
        else right=arr[n-1-i];
    }
    for(int i=0;i<n;i++)
    {
        if(flag[i])ans.push_back(arr[i]);
    }
    cout<<ans.size()<<endl;
    if(ans.size()!=0)
    {
        for(vector<int>::iterator it=ans.begin();it<ans.end();it++)
        {
            if(it==ans.begin())cout<<*it;
            else cout<<" "<<*it;
        }
    }
    else
        cout<<endl;
    
    return 0;
}