·연습 문제

가장 무거운 두 돌

문제 설명PROBLEM

가장 무거운 두 돌

왜 이 문제를 푸나요?

‘가장 무거운 두 돌’은 힙의 핵심 연산을 다른 입력 조건에서도 스스로 선택할 수 있게 만드는 반복 문제입니다. 결과뿐 아니라 경계값과 목표 복잡도까지 확인하세요.

익혀야 할 개념

  • 우선순위 큐
  • heapq
  • 최솟값 선택
  • O(log n) 갱신

과제

매번 가장 무거운 두 돌을 꺼내 무게 차이가 있으면 다시 넣습니다. 마지막 무게 또는 0을 반환하세요.

함수

last_stone_weight(stones)

목표 복잡도

시간 O(n) 이내

테스트 케이스

실행하면 아래 순서대로 채점합니다. print() 출력도 이 순서대로 쌓입니다.

#이름stones기대값
1대표[2, 7, 4, 1, 8, 1]1
2같은 둘[3, 3]0
3한 개[5]5
4빈 목록[]0
5연쇄 충돌[10, 7, 4, 2, 1]0

Python 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.

개념 노트와 실수 포인트

  • 최솟값을 O(1)에 확인하고 O(log n)에 추가·삭제하는 완전이진트리 구조입니다.
  • 파이썬 heapq는 최소 힙입니다.