题目是pat乙级
思路:主元素只要满足左边不大于它,右边不小于它。因此判断一个元素是不是主元素,只要查询左边的最大值和右边的最小值,然后判断就可以了。初看可以使用线段树,而实际上子问题并没有那么多,因为求最大值集合的左边界是固定的,求最小值集合的右边界是固定的。程序中直接定义两个数组保存所有最值,几次遍历就行了。
训练目的:类似DP 数组遍历
思路:主元素只要满足左边不大于它,右边不小于它。因此判断一个元素是不是主元素,只要查询左边的最大值和右边的最小值,然后判断就可以了。初看可以使用线段树,而实际上子问题并没有那么多,因为求最大值集合的左边界是固定的,求最小值集合的右边界是固定的。程序中直接定义两个数组保存所有最值,几次遍历就行了。
训练目的:类似DP 数组遍历
题目描述
著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的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
参考代码
#include <iostream> using namespace std; int data[100010]; int preMax[100010]; int nextMin[100010]; int main() { int n; while(cin>>n) { int i; for(i=0;i<n;i++) cin>>data[i]; preMax[0]=data[0]; for(i=1;i<n;i++) preMax[i]=(preMax[i-1]>data[i]?preMax[i-1]:data[i]); nextMin[n-1]=data[n-1]; for(i=n-2;i>=0;i--) nextMin[i]=(nextMin[i+1]<data[i]?nextMin[i+1]:data[i]); int cc=0; for(i=0;i<n;i++) if (preMax[i]<=data[i]&&data[i]<=nextMin[i]) cc++; cout<<cc<<endl; bool first=true; for(i=0;i<n;i++) if (preMax[i]<=data[i]&&data[i]<=nextMin[i]) if (first) cout<<data[i],first=false; else cout<<" "<<data[i]; cout<<endl; } return 0; }