[BOJ] #11004_K번째 수(파이썬, Python3)

본 게시글은 백준 저지 알고리즘 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.

티어

Silver V

문제

수 N개 A1, A2, ..., AN이 주어진다. A를 오름차순 정렬했을 때, 앞에서부터 K번째 있는 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N(1 ≤ N ≤ 5,000,000)과 K (1 ≤ K ≤ N)이 주어진다.

둘째에는 A1, A2, ..., AN이 주어진다. (-109 ≤ Ai ≤ 109)

출력

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

예제 입력 및 출력 1

5 2
4 1 2 3 5

2

---

풀이

import sys
input = sys.stdin.readline

N, K = map(int, input().split())
A_list = list(map(int, input().split()))

A_list.sort()
print(A_list[K-1])

 

간단하게 정렬하는 문제이다. 파이썬은 sort() 함수가 있기 때문에 이를 이용해주면 오름차순으로 정렬이 된다. 그 후 몇 번째라고 했으니 K-1번째를 뜻한다.