AtCoder Beginner Contest 115 C - Christmas Eve【Python】

https://atcoder.jp/contests/abc115/tasks/abc115_c
AtCoder ProblemsのRecommendationで Difficulty: 291、Solve Probability: 56%でした。
昨日取り組んだ
https://atcoder.jp/contests/abc210/tasks/abc210_c
と比べるとかなりシンプルに解くことができました。
iに対してK個の中からmax,minを毎回求めるのは実行時間制限に間に合わないというのは上記の問題で学びました。

N, K = map(int, input().split())
H = []
for i in range(N):
    H.append(int(input()))
H = sorted(H)
ans = float('inf')
for i in range(N):
    if i+K-1 < N:
        ans = min(ans, H[i+K-1]-H[i])
print(ans)