10. 최소힙 _ 파이썬

전체코드

import sys
import heapq as hq
# 입력 값
# 5
# 3
# 6
# 0
# 5
# 0
# 2
# 4
# 0
# -1
sys.stdin=open("C:\\Users\\input.txt",'rt')

a= []
while True:
    n = int(input())
    if n == -1:
        break
    if n == 0:
        if len(a) == 0:
            print(-1)
        else:
            print(hq.heappop(a))
    else:
        hq.heappush(a, n)

---

이 코드는 최소 힙 (Min Heap) 을 이용하여 입력된 값을 관리하는 프로그램이야.
하나씩 차근차근 설명해줄게! 😊

---

✅ 코드 분석

1️⃣ import heapq as hq

 

import heapq as hq

• heapq 모듈을 hq라는 별칭으로 가져옴
• 파이썬의 heapq는 기본적으로 최소 힙 (Min Heap) 을 제공함

---

2️⃣ 파일 입력 받기

 

sys.stdin=open("C:\\input.txt",'rt')

• "input.txt" 파일에서 입력을 읽어옴
• 코딩 테스트 환경에서는 주석 처리하고 사용해야 함!

---

3️⃣ 초기화

 

a = [] # 최소 힙으로 사용할 리스트

• 빈 리스트 a를 선언
• heapq 모듈을 이용해서 최소 힙으로 사용할 예정

---

4️⃣ 반복문으로 숫자 입력 받기

while True:
	n = int(input()) # 정수 입력

• 입력을 계속 받아서 처리함

---

5️⃣ 종료 조건 (-1 입력 시 종료)

if n == -1:
	break

• -1이 입력되면 프로그램을 종료

---

6️⃣ n == 0일 때 (힙에서 최솟값 제거 및 출력)

 

if n == 0:
    if len(a) == 0:   # 힙이 비어 있으면 -1 출력
        print(-1)
    else:
        print(hq.heappop(a))  # 최솟값을 꺼내서 출력

• 입력된 값이 0이면:
  1. 힙이 비어 있다면 -1 출력
  2. 비어 있지 않으면 heapq.heappop(a)를 사용해 최솟값을 꺼내고 출력

---

7️⃣ 숫자가 0이 아닐 때 (힙에 추가)

 

else:
    hq.heappush(a, n)  # 힙에 새로운 값 추가

• 입력된 값이 0도 아니고 -1도 아니면, 힙에 추가

---

✅ 실행 예제 분석

입력값:

 

5 3 6 0 5 0 2 4 0 -1

 

5	5 추가	[5]
3	3 추가	[3, 5]
6	6 추가	[3, 5, 6]
0	최솟값 3 제거	[5, 6]	3
5	5 추가	[5, 6, 5]
0	최솟값 5 제거	[5, 6]	5
2	2 추가	[2, 6, 5]
4	4 추가	[2, 4, 5, 6]
0	최솟값 2 제거	[4, 6, 5]	2
-1	종료

출력 결과:

3 5 2

---

✅ 정리

• 최소 힙 (heapq)를 이용하여 입력값을 관리
• n == 0이면 힙에서 최솟값을 제거하고 출력
• n == -1이면 프로그램 종료
• 0이 아닌 정수는 힙에 추가