题解 | #最小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


全部评论

相关推荐

真tmd的恶心,1.面试开始先说我讲简历讲得不好,要怎样讲怎样讲,先讲背景,再讲技术,然后再讲提升多少多少,一顿说教。2.接着讲项目,我先把背景讲完,开始讲重点,面试官立即打断说讲一下重点,无语。3.接着聊到了项目的对比学习的正样本采样,说我正样本采样是错的,我解释了十几分钟,还是说我错的,我在上一家实习用这个方法能work,并经过市场的检验,并且是顶会论文的复现,再怎么不对也不可能是错的。4.面试官,说都没说面试结束就退出会议,把面试者晾在会议里面,丝毫不尊重面试者难受的点:1.一开始是讲得不好是欣然接受的,毕竟是学习。2.我按照面试官的要求,先讲背景,再讲技术。当我讲完背景再讲技术的时候(甚至已经开始蹦出了几个技术名词),凭什么打断我说讲重点,是不能听出人家重点开始了?这也能理解,每个人都有犯错,我也没放心上。3.我自己做过的项目,我了解得肯定比他多,他这样贬低我做过的项目,说我的工作是错误的,作为一个技术人员,我是完全不能接受的,因此我就和他解释,但无论怎么解释都说我错。凭什么,作为面试官自己不了解相关技术,别人用这个方式work,凭什么还认为这个方法是错的,不接受面试者的解释。4.这个无可厚非,作为面试官,不打招呼就退出会议,把面试者晾着,本身就是有问题。综上所述,我现在不觉得第一第二点也是我的问题,面试官有很大的问题,就是专门恶心人的,总结面试官说教,不尊重面试者,打击面试者,不接受好的面试者,技术一般的守旧固执分子。有这种人部门有这种人怎么发展啊。最后去查了一下,岗位关闭了。也有可能是招到人了来恶心人的,但是也很cs
牛客20646354...:招黑奴啊,算法工程师一天200?
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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