BOJ 11004 K번째 수

https://www.acmicpc.net/problem/11004
시간 2초, 메모리 512MB
input :

• N K(1 ≤ N ≤ 5,000,000)(1 ≤ K ≤ N)
• A1, A2, ..., AN이 주어진다. (-10^9 ≤ Ai ≤ 10^9)

output :

• A를 정렬했을 때, 앞에서부터 K번째 있는 수를 출력

조건 :

• A를 오름차순 정렬했을 때, 앞에서부터 K번째 있는 수

오름차순 정렬 하라고 했으니, sorted()를 쓰고.
앞에서 부터 K번째 존재하는 수이기 때문에 idx로 k - 1 해서 찾아주었다.

import sys

n, k = map(int, sys.stdin.readline().split())
data = list(map(int, sys.stdin.readline().split()))
data = sorted(data)
print(data[k - 1])

딱 봐도 오래 걸릴거 같아서 pypy로 냈는데도 느리다.. ㄴㅇㄱ