모각코처음인사람들 4회차 2022.01.18

less than 1 minute read

목표

백준 2문제 풀기
인턴 마무리

나는야 포켓몬 마스터 이다솜: 백준

https://www.acmicpc.net/problem/1620

풀이코드

import sys
M,N = map(int,input().split(" "))
numDict = {}
strDict = {}
for i in range(M):
    str = sys.stdin.readline().strip()
    numDict[i+1] = str
    strDict[str] = i+1


# print(numDict)
# print(strDict)
for _ in range(N):
    inputStr = sys.stdin.readline().strip()
    if inputStr.isdigit():
        #숫자이면
        print(numDict[int(inputStr)])
    else:
        print(strDict[inputStr])

최소 힙: 백준

https://www.acmicpc.net/problem/1927

풀이코드

import heapq,sys
N = int(input())
heap = []
for _ in range(N):
    input = int(sys.stdin.readline().strip())
    if input == 0:
        try:
            n = heapq.heappop(heap)
            print(n)
        except IndexError:
            print(0)

    else:
        heapq.heappush(heap,input)