[프로그래머스] Python - 더 맵게

코딩테스트 연습 - 더 맵게

 

정렬을 이용하는 문제다.

최소 힙을 사용해서 지속적으로 최소값 2개를 빼오고 그 결과를 다시 넣는 형태로 구현했다.

여기까지의 구현은 정말 누구나 생각할 수 있고 쉬운 난이도인데 -1을 출력하는 조건에서 살짝 고민을 했다.

K 이상의 맵기를 만들 수 없는 상황이면 2개의 최소값을 뽑아 올 수 없는 상황이라고 생각하고 여러 번 조건을 바꿔봐도 마지막 TC에서 오답처리 되길래 다른 사람의 코드를 보니 Index Error로 예외처리를 해주는 것을 확인했다.

GitHub - Floodnut/algorithm: 알고리즘 풀이 모음