1045. 快速排序

著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元,通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的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

解题思路:

题目看上去很简单,但给出的 N 很大,如果只是简单的按照其规则进行比较,一定会超时。

先导:
将给出的数字复制到另一个数组 (temp) 里,进行排序:

排列前:1 3 2 4 5
排列后:1 2 3 4 5

排列后的数组中,每一个数字大于前面的数字,小于后面的数字。这跟主元的特征是一致的,这表明,如果一个数是被选取的主元,那么它在数组中的位置在排序后不会发生变化。那么 1 4 5 便有可能是主元,但不一定就是主元。比如:

排列前:1 2 5 4 3
排列后:1 2 3 4 5

这里面的 4 就不是主元。说明原位置与排序后的位置相同是其必要而不充分条件。

详细思路:

概要:先将数组排序,然后从最大值开始逆序排除非主元数。如果最大值不在末尾,那么最大值及它之后的元素都不可能是主元,排除这些元素,缩短了序列长度,再看下一个最大值,继续往下走。如果最大值在末尾且其原始位置和排序后的位置一致则是主元,记录下来。重复上面的判断。

多写出几组数据,就可以发现有这样一个规律:

比如原始数组:

1 3 5 2 4 6 9 8 7

顺序排列后:

1 2 3 4 5 6 7 8 9

先看原始数组,其中 9 是最大值,9 如果不在末尾,在 9 及 9 后面的元素 9 和 7 都不可能是主元。于是我们划掉 9、8、7:

1 3 5 2 4 6 9 8 7

剩下的值最大的为 6。6 后面没有数字,说明 6 有可能是主元。

如果 6 是主元,那么 6 的前面应有 5 个比它小的元素,后面应有 3 个比它大的元素。顺序排列后的数组中的 6 的位置就满足这一要求。

6 所在位置(数组下标)是 5,与按顺序排列后的位置(排列后的 6 数组下标也为 5) 一致,这个时候,可以判断 6 一定是主元元素。因为它与排列后的 6 的位置一致,说明前面一定有 5 个元素,而它又是当前序列的最大值,说明这个 6 比前面的 5 个元素都大,因而也可推知比它后面的元素都小,因为 6 的前面只可能有 5 个元素,而这 5 个又都比它小,说明大的都在它之后。

这样就得到了一个主元 6,存入 target 数组中。再删去后面 4 个元素:

1 3 5 2 4 6 9 8 7

接下来,最大值是 5,5 的后面有2 4,这样 5、2、4 就不可能是主元。就再删去 5 2 4

1 3 5 2 4 6 9 8 7

接下来,最大值 3,与顺序排列后 3 的位置不一致,肯定不是主元。删掉 3。继续处理:

1 3 5 2 4 6 9 8 7

接下来,最大值是 1,与顺序排列后的位置一致,则它是主元。于是我们得到了所有结果。

[后记]
搜索了其它童鞋的思路,发现只要与顺序排列后的数组比较就可以了,只需要额外添加一个变量记录从 0 到当前位置的最大值 max。如果正在处理的数与顺序排列后的一致,又等于这个最大值,说明它就是主元。这样的代码更简单些,无需使用结构来记录原编号了。

这篇代码也不算复杂,不过多个思路总是好些。

😃

解题代码:

#include<stdio.h> 
#include<stdlib.h>

int compare (const void *p, const void *q);

struct array {
	int number;
	int id; // 记录在原始数组中的位置 
}; 

int main () 
{	 
	int N;
	scanf ("%d\n", &N);
	struct array a[N], temp[N];
	for (int i=0; i<N; i++) {
		scanf ("%d", &a[i].number);
		temp[i].number = a[i].number;
		a[i].id = i;
		temp[i].id = i;  
	}
	qsort (temp, N, sizeof (struct array), compare);
	
	int count = 0;
	int target[N];
	int temp_N = N - 1; // 初始化为数组最后一个元素下标 
	for (int i=N-1; i>=0; i--) {
		if (temp[i].id < temp_N) {
			temp_N = temp[i].id - 1; // 如果这个最大值不在末尾,将 N 的范围减小至这个数的前一位 
		} else if (temp[i].id == temp_N) {
			if (temp[i].id == i) {
				target[count++] = temp[i].number;
			} // 如果原位置等于排列后的位置,则是主元 
			temp_N = temp[i].id - 1;
		}
	}
	printf ("%d\n", count);
	if (count) {
		for (int i=count-1; i>0; i--) {
			printf ("%d ", target[i]);
		}
		printf ("%d\n", target[0]);
	} else {
		printf ("\n"); // 0 个不输出,但要输出回车 
	}
	
	return 0;
} 

int compare (const void *p, const void *q) {
	int x = (*(struct array *)p).number;
	int y = (*(struct array *)q).number;
	if (x > y) {
		return 1;
	} else if (x < y) {
		return -1;
	}
	return 0;
}
posted @ 2016-09-01 21:33  文之  阅读(919)  评论(0编辑  收藏  举报