BOJ 11004 K番号

https://www.acmicpc.net/problem/11004
2秒、512 MBメモリ
input :

N K(1 ≤ N ≤ 5,000,000)(1 ≤ K ≤ N)

A1, A2, ..., ANが与えられた.(-10^9 ≤ Ai ≤ 10^9)

output :
並べ替え

Aの場合、前からK番目の数字

が出力.
条件:

Aを昇順に並べると、前からK番目の数字は

となる.
昇順で並べ替えてもらうので、並べ替え()を書きます.
前から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])

見るからに時間がかかるので、Ppyで発行するのも遅いです.推測を表す