题解 | #最小K个值#

最小的K个数

http://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf

利用最小堆解决问题

def left(i):
	""" 求节点i的左子树 """
	return 2 * i + 1


def right(i):
	""" 求节点i的右子树,在数组中,右子树比左子树大1 """
	return left(i) + 1


def exchange(arr, i, j):
	""" 交换数组中两个位置的值 """
	temp = arr[i]
	arr[i] = arr[j]
	arr[j] = temp


def heapify(arr, i, size):
	""" 维护最小堆 """
	l, r, small = left(i), right(i), 0
	# small保存较小的那个值
	if l < size and arr[l] < arr[i]:
		small = l
	else:
		small = i
	if r < size and arr[r] < arr[small]:
		small = r
	if small != i:
		# 当前节点与较小的值
		exchange(arr, i, small)
		# 递归进行维护堆
		heapify(arr, small, size)


def make_heap(arr):
	""" 构建一个最小堆 """
	alen = len(arr)
	for i in range(alen // 2, -1, -1):
		heapify(arr, i, alen)


class Solution:
	def GetLeastNumbers_Solution(self, tinput: list, k):
		make_heap(tinput)
		alen = len(tinput)
		res = []
		for i in range(k):
			# 从最小堆中取出第一个节点的值,这个值是最小值
			res.append(tinput[0])
			# 将当前节点废弃,
			exchange(tinput, 0, alen - 1)
			alen -= 1
			# 重新维护堆
			heapify(tinput, 0, alen)
		return res


全部评论

相关推荐

02-06 16:33
门头沟学院 Java
Phoenix_Re...:很少最多一两个,大多数保研或者摆烂等着考研
点赞 评论 收藏
分享
HR_丸山彩同学:你的项目描述里,系统设计讲了很多:MemCube是什么、三级存储架构怎么设计、四种遗忘策略分别是什么。这些面试的时候讲没问题,但简历上不需要这么细。 简历要突出的是影响力,不是实现细节。面试官看简历的时候想知道的是「这个项目有多大价值」,不是「这个项目具体怎么实现的」。实现细节是面试时候聊的 怎么改:技术细节可以精简为一句「采用三级存储架构+四种遗忘策略」,把省出来的篇幅用来写影响力。比如:项目有没有开源?有没有写成技术博客?有没有被别人使用过? 校园经历没有任何信息量,任何人都可以写这句话,写了等于没写。更关键的是,你投的是技术岗,校园活动经历本来就不是加分项。如果非要写,必须写出具体的数字和成果。如果你没有这些数字,那就老老实实删掉 「端到端耗时缩减30-40%」要给出确切数字和绝对值。从1000ms降到600ms是降了40%,从100ms降到60ms也是降了40%,但这两个含义完全不一样。其他也是,涉及到数据,准备好证据,口径统一,面试会问 「熟练」「熟悉」「了解」混在一起用,读起来很乱。而且「了解前端需求」最好改成「具备前后端协作经验」
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务