알고리즘/문제풀기
[백준/파이썬] K번째 수 11004
hyun_jo_o
2019. 7. 9. 23:09
문제
수 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이용해서 입력 받기 참고