PAT 乙级练习 1045 快速排序

PAT 乙级练习 题解合集

本题链接

题目

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

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

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

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

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

输入样例：

5
1 3 2 4 5

输出样例：

3
1 4 5

思路

1. 用长度为n的nums数组接收数据；
2. 定义长度为n的primary数组。若primary[i] == 1表示nums[i]可能是主元，将其所有元素先初始化为1；
3. 从左到右遍历nums数组，用来排除掉那些左边元素比它大的元素，这些元素不可能是主元。
   具体做法：
   a. 先定义max等于一个足够小的数，比如0；
   b. 从左到右开始遍历，如果nums[i] > max，更新max的值，如果nums[i] > max，则不符合条件，primary[i] = 0。注意不用考虑相等的情况，因为题目说所有数字互不相等。
4. 再从右到左遍历，排除掉那些右边元素比它小的元素，做法类似 3 ，min的初始值要足够大，比如1000000007。

---

坑点：

• 测试点2格式错误，因为没有主元素的时候输出应当是0\n\n，有两个换行。

代码

#include <stdio.h>

int main() {
	int n, i;
	scanf("%d", &n);
	int nums[n], primary[n], cnt = 0;
	for (i = 0; i < n; ++i)
		primary[i] = 1;
	for (i = 0; i < n; ++i)
		scanf("%d", &nums[i]);
	int max = 0, min = 1000000007;
	for (i = 0; i < n; ++i)	{
		if (nums[i] > max)
			max = nums[i];
		else
			primary[i] = 0;
	}
	for (i = n - 1; i >= 0; --i) {
		if (nums[i] < min)
			min = nums[i];
		else
			primary[i] = 0;
	}
	for (i = 0; i < n; ++i)
		if (primary[i])
			++cnt;
	printf("%d\n", cnt);
	int flag = 0;
	if (cnt) {
		for (i = 0; i < n; ++i)
			if (primary[i]) {
				if (flag)
					putchar(' ');
				flag = 1;
				printf("%d", nums[i]);
			} 	
	}
	putchar('\n'); 
	return 0;
}