第七周作业(北大选课同学选做)
题目
1.快速排序主元(10分)
题目内容:著名的快速排序算法里有一个经典的划分过程:我们通常采用某种方法取一个元素作为主元(中值),通过交换,把比主元小的元素放到它的左边,比主元大的元素放到它的右边。 给定划分后的N个互不相同的正整数的排列,请问有多少个元素可能是划分前选取的主元?
例如给定的排列是[1, 3, 2, 4, 5]。则:1 的左边没有元素,右边的元素都比它大,所以它可能是主元;尽管 3 的左边元素都比它小,但其右边的 2 比它小,所以它不能是主元;尽管 2 的右边元素都比它大,但其左边的 3 比它大,所以它不能是主元;类似原因,4 和 5 都可能是主元。因此,有 3 个元素可能是主元。
输入格式:
一行数个整数的排列,由空格分隔
输出格式:
在第 1 行中输出有可能是主元的元素个数;在第 2 行中按递增顺序输出这些元素,其间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:1 3 2 4 5
输出样例:
3
1 4 5
时间限制:500ms 内存限制:32000kb
思路
本头脑简单想出的头脑简单思路,极有可能时间或内存超出限制(记录完这篇就去看看大佬的思路)。判断一个数是否主元,如果这个数左边的数都比他小,并且右边的数都比他大,他就是主元。
即:
条件1:左边这部分列表的最大值,小于等于这个数,(如果左边没有数值则默认其满足条件1)
条件2:右边这部分列表的最小值,大于等于这个数,(如果右边没有数值则默认满足条件2)
使用if的or语句判断条件时,把判断列表是否为空写在or前面。如果列表空,就不再判断or后面的条件。因为max()执行的对象是空列表,就会报错。
代码
def judge_mid(blist):
#将输入的字符串转换成数字列表
alist = []
for i in blist.split():
alist.append(int(i))
Mid_list = []
#比较左右数值判断主元,Mid_list列表记录主元
for position in range(len(alist)):
if alist[:position]==[] or max(alist[:position])<=alist[position]:
if alist[position+1:]==[] or min(alist[position+1:])>=alist[position]:
Mid_list.append(alist[position])
print(len(Mid_list))
#升序排序、将列表转换数字字符串输出
Mid_list.sort()
for i in Mid_list:
print(i, end=' ')
blist = input()
judge_mid(blist)
欢迎指正