PAT B1045 快速排序 (25分)

著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列，请问有多少个元素可能是划分前选取的主元？

例如给定 $N=5$, 排列是1、3、2、4、5。则：

• 1 的左边没有元素，右边的元素都比它大，所以它可能是主元；
• 尽管 3 的左边元素都比它小，但其右边的 2 比它小，所以它不能是主元；
• 尽管 2 的右边元素都比它大，但其左边的 3 比它大，所以它不能是主元；
• 类似原因，4 和 5 都可能是主元。

因此，有 3 个元素可能是主元。

输入格式：

输入在第 1 行中给出一个正整数 N（≤10​⁵​​）； 第 2 行是空格分隔的 N 个不同的正整数，每个数不超过 10​^9​​。

输出格式：

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

输入样例：

5
1 3 2 4 5

输出样例：

3
1 4 5

• 思路 1：
  一个元素如果前面所有元素都比他小，后面都比他大，那他一定处于最终位置！
  题目要求找这样的元素，显然这种元素一定处于最终位置，（否则一定不满足左小右大，总共就那些元素，他不在自己位置上，肯定挤走了别人）
  所以，可以先sort得到每个 元素最终的位置，不在最终位置的元素一定不是答案（筛选掉一部分），但是在最终位置上的元素就一定是答案吗？（显然 NO~）如（123456对于4将前后随机调换几个元素后156432，4还在最终位置，但不是所求元素）
  所以只有处于最终位置这一个条件是不够的，但如果一个元素处于最终位置，并且他左边都比他小，那他一定是所求元素（对每个元素来说，比他小的就那些，即排好序后前面的所有，所以如果一个元素处在最终位置，前面有全比他小，后面不可能有比他小的了，都在前面）

->总结：

1. 先sort
2. 在从头遍历判断 1)是否处于最终位置 2)前面是否有比他大的（通过不断更新全局Max->前驱序列中最大元素）

• code 1:

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
int num[100010], tmp[100010], ans[100010];
int main(){
	int n, Max = 0, cnt = 0;
	scanf("%d", &n);
	for(int i = 0; i < n; ++i){
		scanf("%d", &num[i]);
		tmp[i] = num[i];
	}
	sort(tmp, tmp+n);
	for(int i = 0; i < n; ++i){
		if(num[i] == tmp[i] && num[i] > Max){
			ans[cnt++] = num[i];
		}
		Max = max(Max, num[i]);	
	}
	printf("%d\n", cnt);
	for(int i = 0; i < cnt; ++i){
		printf("%d", ans[i]);
		if(i < cnt-1) printf(" ");
	}
	printf("\n");	//处理空元素 
	return 0;
}

• 思路 2：
  打表法，同PAT A1093，用两个数组preMax[]和postMin[]分别记录，对应位置元素前面最大值和后面最小值
  先两次遍历，求出preMax、和postMin
  在对原数组做一次遍历，通过判断每个元素是否比他前面最大的大，后面最小的小，得到最终结果

• code 2:

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
const int INF = 0x3fffffff; 
int num[100010], preMax[100010], postMin[100010], ans[100010];
int main(){
	int n, Max = 0, Min = INF, cnt = 0;
	scanf("%d", &n);
	for(int i = 0; i < n; ++i){
		scanf("%d", &num[i]);
	}
	for(int i = 0; i < n; ++i){
		preMax[i] = Max;
		Max = max(Max, num[i]);	
	}
	for(int i = n-1; i >= 0; --i){
		postMin[i] = Min;
		Min = min(Min, num[i]);
	}	//!!!:优化 1
	for(int i = 0; i < n; ++i){
		if(num[i] < postMin[i] && num[i] > preMax[i])
			ans[cnt++] = num[i]; 
	}
	printf("%d\n", cnt);
	for(int i = 0; i < cnt; ++i){
		printf("%d", ans[i]);
		if(i < cnt-1) printf(" ");
	}
	printf("\n");	//处理空元素 
	return 0;
}

• 优化 1：

//求最大最小数组可以简写为：
preMax[0] = 0;	//记录最大值
for(int i = 1; i < n; ++i){
	preMax[i] = max(preMax[i-1], num[i-1]);
}

• T2 code：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 110000;
int in[maxn], tmp[maxn];
vector<int> ans;

int main(){
	int n;
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i){
		scanf("%d", &in[i]);
		tmp[i] = in[i];
	}
	sort(tmp, tmp+n);
	int Max = 0;
	for(int i = 1; i <= n; ++i){
		Max = max(Max, in[i]);
		if(in[i] == tmp[i] && in[i] == Max){
				ans.push_back(in[i]);
		}
	} 
	sort(ans.begin(), ans.end());
	printf("%d\n", ans.size());
	for(int i = 0; i < ans.size(); ++i){
		if(i == 0) printf("%d", ans[i]);
		else printf(" %d", ans[i]);
	}
	printf("\n");
}