졍's 기술 블로그

N번째 큰 수

티어 : Gold 5
시간 제한 : 1 초
메모리 제한 : 12 MB
알고리즘 분류 : 자료 구조, 정렬, 우선순위 큐

문제

N×N의 표에 수 N2개 채워져 있다. 채워진 수에는 한 가지 특징이 있는데, 모든 수는 자신의 한 칸 위에 있는 수보다 크다는 것이다. N=5일 때의 예를 보자.

이러한 표가 주어졌을 때, N번째 큰 수를 찾는 프로그램을 작성하시오. 표에 채워진 수는 모두 다르다.

입력

첫째 줄에 N(1 ≤ N ≤ 1,500)이 주어진다. 다음 N개의 줄에는 각 줄마다 N개의 수가 주어진다. 표에 적힌 수는 -10억보다 크거나 같고, 10억보다 작거나 같은 정수이다.

출력

첫째 줄에 N번째 큰 수를 출력한다.

예제 입출력

Algorithm

정렬
1. 입력받을 때마다 내림차순 정렬해 리스트에 N개 씩 저장
2. 입력을 모두 받았을 때는 리스트의 가장 마지막 값 출력

Code

import sys
input = sys.stdin.readline
N = int(input())
nums = []
for _ in range(N):
    nums += list(map(int, input().split()))
    nums.sort(reverse = True)
    nums = nums[:N]

print(nums[-1])

메모리: 30864 KB
시간: 844 ms