<div id="readability-page-1" class="page"><div><h2 id="최대힙">최대힙</h2>
참고자료: <a href="https://www.inflearn.com/course/%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%AC%B8%EC%A0%9C%ED%92%80%EC%9D%B4-%EC%BD%94%EB%94%A9%ED%85%8C%EC%8A%A4%ED%8A%B8/lecture/26952?tab=curriculum&amp;speed=2">파이썬 알고리즘 문제풀이</a>
상기 강의 내용을 바탕으로 swift 코드로 바꿔서 요약 정리한 내용입니다.
<ul>
<li>문제</li>
</ul>
(1). 자연수가 입력되면 최대힙에 입력, (2). 0이 입력되면 최대값을 꺼내어 출력(값이 없으면 -1을 출력), (3). -1이 입력되면 프로그램이 종료되도록 프로그램을 작성하시오.
<ul>
<li>입력예제</li>
</ul>
5 
3 
6 
0 
5 
0 
2 
4 
0 
-1
<ul>
<li>출력예제</li>
</ul>
6 
5 
5
<h2 id="내-풀이">내 풀이</h2>
<pre><code>var a = [Int]()
while true {
 let n = Int(readLine()!)!
 if n == -1 { break }
 if n == 0 {
 if a.isEmpty {
 print(-1)
 } else {
 print(-a.popLast()!)
 }
 } else {
 a.insert(-n, at: 0)
 a.sort(by: &gt;)
 }
}</code></pre>
최소힙 코드를 그대로 유지하고 값을 넣고, 뺄때에만 - 부호를 붙여주면 최대힙을 만들 수 있다.
<h2 id="강의-풀이">강의 풀이</h2>
별도 풀이가 존재하지 않기 때문에 파이썬 코드만 살펴보겠다.
<h2 id="파이썬-코드">파이썬 코드</h2>
<pre><code>import heapq as hq
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)</code></pre></div></div>

PS & Algorithm

WooSeok Suh

최대힙
참고자료: 파이썬 알고리즘 문제풀이
상기 강의 내용을 바탕으로 swift 코드로 바꿔서 요약 정리한 내용입니다.
문제
(1). 자연수가 입력되면 최대힙에 입력, (2). 0이 입력되면 최대값을 꺼내어 출력(값이 없으면 -1을 출력), (3). -1이 입력되면 프로그램이 종료되도록 프로그램을 작성하시오.
입력예제
5
3
6
0
5
0
2
4
0
-1
출력예제
6
5
5
내 풀이
var a = [Int]()
while true {
    let n = Int(readLine()!)!
    if n == -1 { break }
    if n == 0 {
        if a.isEmpty {
            print(-1)
        } else {
            print(-a.popLast()!)
        }
    } else {
        a.insert(-n, at: 0)
        a.sort(by: >)
    }
}
최소힙 코드를 그대로 유지하고 값을 넣고, 뺄때에만 - 부호를 붙여주면 최대힙을 만들 수 있다.
강의 풀이
별도 풀이가 존재하지 않기 때문에 파이썬 코드만 살펴보겠다.
파이썬 코드
import heapq as hq
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)

자료구조 활용 - 힙(2)

댓글