PAT (Basic Level) Practice 1045

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

分析：主元需要满足的条件是:左边的数都比自己小，右边的数都比自己大。这时可以采用升序排序观察的方法，因为在排序之后，每个数都满足左边的数都比自己小，右边的数都比自己大的条件。于是可以找出前后位置不变的项作为备选答案。同时，为了保证在排序前数也满足条件，记录每个数之前的最大值与之比较，不大于最大值的才满足条件。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
#define MAXN 100001
using namespace std;
int num1[MAXN] = { 0 };
int num2[MAXN] = { 0 };
bool isChecked[MAXN];
vector<int> v;
int main() {
	int N;
	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> num1[i];
		num2[i] = num1[i];
	}
	sort(num1, num1 + N);
	int maxN = -1;
	for (int i = 0; i < N; i++) {
		if (num1[i] == num2[i] && num1[i] > maxN) {
			v.push_back(num1[i]);
		}
		if (maxN < num2[i]) {
			maxN = num2[i];
		}
	}
	cout << v.size() << endl;
	if (v.size() == 0) {
		cout << endl;
	} else {
		for (int i = 0; i < v.size(); i++) {
			if (i != v.size() - 1) {
				cout << v[i] << ' ';
			} else {
				cout << v[i] << endl;
			}
		}
	}
	return 0;
}