사용자 도구

사이트 도구


ps:problems:boj:1715

카드 정렬하기

ps
링크acmicpc.net/…
출처BOJ
문제 번호1715
문제명카드 정렬하기
레벨골드 4
분류

그리디, 우선순위큐

시간복잡도O(nlogn)
인풋사이즈n<=100,000
사용한 언어Python
제출기록33692KB / 196ms
최고기록124ms
해결날짜2021/12/13
태그

[라이] 우선순위 큐

풀이

  • 그리디한 방식으로 가장 작은 묶음 두개를 골라서 합치는 것을 반복하는 것이 최적이다.
  • 가장 작은 묶음을 꺼내고, 새 묶음을 추가하는 작업은 우선순위 큐 (Priority Queue)를 통해서 처리하면 된다. pop과 push 모두 O(logn)이 걸린다. 이렇게 두 묶음을 합치는 작업을 n-1번 해야지 한묶음이 되므로, 총 시간복잡도는 O(nlogn)

코드

"""Solution code for "BOJ 1715. 카드 정렬하기".

- Problem link: https://www.acmicpc.net/problem/1715
- Solution link: http://www.teferi.net/ps/problems/boj/1715

Tags: [Priority queue]
"""

import heapq
import sys


def main():
    N = int(sys.stdin.readline())
    size_heap = [int(sys.stdin.readline()) for _ in range(N)]

    heapq.heapify(size_heap)
    answer = 0
    for _ in range(N - 1):
        s1 = heapq.heappop(size_heap)
        s2 = heapq.heappop(size_heap)
        answer += s1 + s2
        heapq.heappush(size_heap, s1 + s2)        

    print(answer)


if __name__ == '__main__':
    main()

토론

댓글을 입력하세요:
M O A P J
 
ps/problems/boj/1715.txt · 마지막으로 수정됨: 2022/07/05 16:13 저자 teferi