티스토리 뷰
728x90
반응형
https://www.acmicpc.net/problem/11279
내 풀이
import sys
import heapq
arr = []
for _ in range(int(sys.stdin.readline())):
x = int(sys.stdin.readline())
#(iterator,(우선순위, 넣을 값))
if x!=0:
heapq.heappush(arr,(-x,x))
else:
if arr:
num = heapq.heappop(arr)
print(num[1])
else:
print(0)
heapq의 heappush는 기본적으로 최소힙으로 이루어져있기 때문에 최대 힙을 구현하기 위해서는 우선순위도 지정해주어야 한다.
따라서, 우선순위를 -x로 지정하면 최대힙 구현이 가능하다.
힙큐 관련 자료 (시간복잡도)
https://velog.io/@sossont/Python-heapq-%EB%AA%A8%EB%93%88
다른 사람 풀이
import io
import os
import sys
import heapq
input = io.BytesIO(os.read(0, os.fstat(0).st_size)).readline
def solve():
q = []
res = []
for _ in range(int(input())):
if (a := int(input())):
heapq.heappush(q, -a)
else:
res.append(-heapq.heappop(q) if q else 0)
sys.stdout.write('\n'.join(map(str, res)))
if __name__ == '__main__':
solve()
정수인지 판별할 때 이런 코드를 쓸 수 있다는 것도 꺠달았다.
반응형
'알고리즘 > Baekjoon' 카테고리의 다른 글
[CodingTest] 백준 #11286 절댓값 힙 (0) | 2021.12.20 |
---|---|
[CodingTest] 백준 #1927 최소 힙 (0) | 2021.12.20 |
[CodingTest] 백준 #11051 이항 계수 2 (0) | 2021.12.14 |
[CodingTest] 백준 #11050 이항 계수 1 (0) | 2021.12.12 |
[CodingTest] 백준 #1676 팩토리얼 0의 개수 (0) | 2021.12.12 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 브루트 포스
- web
- javascript
- Case When
- html
- jsp
- R
- Django
- 자바스크립트
- 큐
- 장고
- 단계별로풀어보기
- jQuery
- Java
- CSS
- 정렬
- Oracle
- 프로그래머스
- 덱
- 파이썬
- python
- 백준
- bootstrap
- baekjoon
- brute force
- 문자열
- 스프링
- 자바
- append
- 고득점 키트
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | |||
5 | 6 | 7 | 8 | 9 | 10 | 11 |
12 | 13 | 14 | 15 | 16 | 17 | 18 |
19 | 20 | 21 | 22 | 23 | 24 | 25 |
26 | 27 | 28 | 29 | 30 | 31 |
글 보관함