题目大意:给出N个不同正数的序列,如果当中的某个数,大于左边所有数,小于右边所有数,则这个数可以被选为主元。输出所有这样的数。
与快排相关,快排的partition操作是将无序序列的首元素放到它应该在的位置,同时使左边的数都小于它,右边的数都大于它。因此,将给出的序列排序后,与原序列位置相同的元素都可能是主元,但还要满足一个条件,就是大于左边的所有数(因为是N个不同的正数,大于左边的所有数,并且又在有序序列的相同位置,那么一定小于右边的所有数),例如 3 2 1 5 4中的2,满足与有序序列位置相同的条件,但是不满足大于左边的所有数。因此,在遍历查找时,需要记录左边的最大值。
一个坑点是如果没有满足条件的元素,输出换行。
AC代码:
#include <vector>
#include <algorithm>
#include <cstdio>
using namespace std;
int main()
{
int N;
scanf("%d", &N);
vector<int> v(N), tmp;
for (int i = 0; i < N; ++i)
scanf("%d", &v[i]);
tmp = v;
sort(tmp.begin(), tmp.end());
int leftMax = 0;
vector<int> ans;
for (int i = 0; i < N; ++i)
{
if(tmp[i] == v[i] && v[i] > leftMax) ans.push_back(v[i]);
if(v[i] > leftMax) leftMax = v[i];
}
printf("%d\n", ans.size());
for (int i = 0; i < ans.size(); ++i)
{
printf("%d", ans[i]);
if(i < ans.size() - 1) printf(" ");
}
if(ans.size() == 0) printf("\n");
return 0;
}