1045 快速排序
著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列,请问有多少个元素可能是划分前选取的主元?
例如给定 N = 5 N = 5 N=5, 排列是1、3、2、4、5。则:
1 的左边没有元素,右边的元素都比它大,所以它可能是主元;
尽管 3 的左边元素都比它小,但其右边的 2 比它小,所以它不能是主元;
尽管 2 的右边元素都比它大,但其左边的 3 比它大,所以它不能是主元;
类似原因,4 和 5 都可能是主元。
因此,有 3 个元素可能是主元。
输入格式:
输入在第 1 行中给出一个正整数 N(N ≤ 105); 第 2 行是空格分隔的 N 个不同的正整数,每个数不超过 109 。
输出格式:
在第 1 行中输出有可能是主元的元素个数;在第 2 行中按递增顺序输出这些元素,其间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:
5
1 3 2 4 5
输出样例:
3
1 4 5
思路:一个元素如果是主元,要求它左边(如果有)元素都比它小,右边(如果有)元素都比它大,即它是序列中到当前位置的最大值。那么最简单的想法就是暴力遍历,看是否符合上述要求,该做法需要两重循环,时间复杂度为O(N2),若 N = 105 则必然超时(限时200ms)。
必须寻求更简便的方法,重新观察主元的要求,左边的元素都比它小,说明即使对整个序列重新排序,主元的位置也不会发生改变(如果改变,要么右边有元素比它小,要么左边有元素比它大,导致在左边或右边新插入了元素,故不是主元)。但这只是必要条件,即排序前后位置不变的元素未必就是主元。
观察序列 3 2 1,排序前后2位置不变,但2不是主元。这是为什么?因为左边的数不都比它小。故若一个数是主元,还需要一个条件:它是原序列中到当前位置的最大值。
综上,得到本题的算法,将原数组a[ ]复制一份到数组b[ ],对b升序排序,并从头遍历b,每次比较a[i]和b[i],若值相同,且为原序列当前位置最大值(a[i] > max),它就是主元。此外,还要随时更新最大值max。由此直到遍历完成。这个算法的时间复杂度源于sort,为O(NlogN),可以通过。
#include <stdio.h>
#include <string.h>
#include <algorithm>
int a[100000] = { 0 }, b[100000] = { 0 }, c[100] = { 0 };
int main() {
int n, cnt = 0, max = 0;
scanf("%d", &n);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
memcpy(b, a, sizeof(a));
std::sort(b, b + n);
for (int i = 0; i < n; i++) {
if (a[i] == b[i] && a[i] > max) c[cnt++] = a[i]; //位置不变,且为原序列当前最大值
if (a[i] > max) max = a[i];
}
std::sort(c, c + cnt);
cnt ? printf("%d\n", cnt) : printf("%d\n\n", cnt); //巨坑,测试点2中cnt=0居然要求输出空行补足
for (int i = 0; i < cnt; i++) {
printf("%d", c[i]);
if (i != cnt - 1) printf(" ");
}
return 0;
}