글 읽기 - 7662 이중우선순위큐 질문 있습니다

daniel0801 2년 전 0

문제랑 우선순위 큐랑 직접적인 관련은 없지만 26번째 줄에서 if valid[-max_pop]: 에서 if valid[-max_pop] == True:로 하면 1보다 큰 값은 포함이 되지 않더라고요. 혹시 두 식의 차이점 알려주실 있으시면 감사하겠습니다!!

import sys
import heapq
input = lambda: sys.stdin.readline().rstrip()
t = int(input())
for _ in range(t):
  k = int(input())
  max_heap = []
  min_heap = []
  valid = {}
  for _ in range(k):
    com, num = input().split()
    num = int(num)
    if com == 'I':
      heapq.heappush(max_heap, -num)
      heapq.heappush(min_heap, num)
      try:
        if valid[num] == True:
          valid[num] += 1
      except KeyError:
        valid[num] = 1
    if com == 'D':
      if num == 1:
        while max_heap:
          max_pop = heapq.heappop(max_heap)
          try:
            if valid[-max_pop]:
              valid[-max_pop] -= 1
              if valid[-max_pop] == 0:
                del valid[-max_pop]
              break
          except KeyError:
            continue
      elif num == -1:
        while min_heap:
          min_pop = heapq.heappop(min_heap)
          try:
            if valid[min_pop]:
              valid[min_pop] -= 1
              if valid[min_pop] == 0:
                del valid[min_pop]
              break
          except KeyError:
            continue
    #print(valid)
  key = list(valid.keys())
  if key:
    key.sort()
    print(key[-1])
    print(key[0])
  else:
    print('EMPTY')