[leetcode] 배열의 K번째 큰 요소

김민서·2024년 1월 6일
0

알고리즘 문제풀이

목록 보기
14/47
  1. Kth Largest Element in an Array
    링크텍스트
from heapq import heappop, heapify

nums = [3,2,1,5,6,4]
k = 2

nums = [i * -1 for i in nums] 
# 최소힙을 최대힙처럼 사용하기 위함
heapify(nums)

for _ in range(k):
	output = heappop(nums) * -1
return output
from heapq import heappush, heappop
nums = [3,2,1,5,6,4]
k = 2
heap = []

# 최소힙을 최대힙처럼 사용하기 위함
for _ in range(k):
	heappush(heap, (-i, i)) 
# [(-6, 6), (-5, 5), (-4, 4), (-5, 5), (-2, 2), (-3, 3), (-3, 3), (-1, 1), (-2, 2)]


for _ in range(k):
	output = heappop(heap)[1]

print(output)

0개의 댓글