赞
踩
题目:给定一个整数数组,如何快速地求出该数组中第k小的数。例:数组arr = [4, 3, 6, 7,1, 3],那么该数组第4小的数为3。
分析:假如该数组为有序,Python通过对数组进行切片就可以找到第k小的数。因此,可以考虑对该数组进行排序,然后再查找出该数。由于只需要找出第k小的数,为了降低是按复杂度,我们采用部分排序方法。该方法的时间复杂度为O(N*k)。并没有开辟新的存储空间。
思路:第一次遍历数组找出最小的数,第二次遍历剩下的数同样找出最小的数(该数即为该数组第二小的数),第k次遍历即可以从剩余len(arr)-k+1个数中找出最小的数(即为该数组中第k小的数)。
代码:
def findSmallKnum(arr, k):
for i in range(k):
for j in range(i + 1, len(arr)):
if arr[i] > arr[j]:
arr[i], arr[j] = arr[j], arr[i]
return arr[k-1]
if __name__ == "__main__":
arr = [4, 3, 6, 7,1, 3]
print(findSmallKnum(arr, 3))
程序运行结果为:3
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。