[백준/파이썬] K번째 수 11004

문제

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

입력

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

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

출력

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

나의 코드

n, k = map(int,input().split())
num = list(map(int, input().split()))		# list형식으로 바로 숫자를 입력받아서 넣음
num.sort()

for i in range(1, n+1):
    if i==k:
        print(num[i-1])

 

map과 sort를 사용해서 쉽게 해결 할 수 있었다. list형식으로 바로 숫자를 받아줘서 저장한 후 sort로 정렬해주고 k번째 수를 구하는 문제였다.

 

 

map이용해서 입력 받기 참고

https://dojang.io/mod/page/view.php?id=2286