사용자 도구

사이트 도구


ps:problems:boj:2075

N번째 큰 수

ps
링크acmicpc.net/…
출처BOJ
문제 번호2075
문제명N번째 큰 수
레벨골드 5
분류

우선순위 큐

시간복잡도O(n^2logn)
인풋사이즈n<=1500
사용한 언어Python
제출기록31524KB / 812ms
최고기록692ms
해결날짜2021/12/14
태그

[라이] 우선순위 큐

풀이

  • 기본적인 방법은 그냥 전체 입력을 다 리스트에 저장해놓고서 정렬하거나 선택알고리즘을 적용하는 것이다. 하지만 이 문제에서는 메모리 제한이 걸려있어서 전체 입력을 리스트에 저장하지 않는 방법이 필요하다.
    • 메모리 제한이 좀 어정쩡한데.. https://www.acmicpc.net/board/view/36675 에 따르면, 원래 문제 의도는 N^2개의 수를 메모리에 모두 담는것을 허용하되 타이트하게 잡는 것이었다고 한다. 그래서 c++ 기준으로는 입력값을 모두 메모리에 저장해서 푸는 것도 가능한것 같은데, 파이썬은 그렇게는 안되는것 같다.
  • 우선순위 큐 - 가장 큰 수 k개 찾기 방법을 적용한다. 처음 N개의 원소로 최소힙을 만들고, 새 원소를 힙에 push하고 최소값을 pop 하는것을 반복한다. 이렇게 하면 힙안의 원소 갯수는 N개로 유지되고, 이 N개의 원소들이 여태까지 봤던 원소들중 에서 가장 큰 원소들이다. N번째로 큰 원소는 당연히 top에 있는 원소. 시간복잡도는 O(N^2logN).

코드

"""Solution code for "BOJ 2075. N번째 큰 수".

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

import heapq
import sys

INF = float('inf')


def main():
    N = int(sys.stdin.readline())
    heap = [-INF] * N
    for _ in range(N):
        nums = [int(x) for x in sys.stdin.readline().split()]
        for num in nums:
            if num > heap[0]:
                heapq.heapreplace(heap, num)
    print(heap[0])


if __name__ == '__main__':
    main()

토론

댓글을 입력하세요:
E᠎ W᠎ H H Z
 
ps/problems/boj/2075.txt · 마지막으로 수정됨: 2022/07/05 16:13 저자 teferi