백준 11004번 - 'K번째 수' (PYTHON 풀이)

문제:

 

A의 개수와 K의 수를 첫째줄에 입력받고, 둘째 줄에 A를 전부 입력받을 때 해당 수를 오름차순으로 정렬한 다음 K에 해당하는 값을 출력하면 되는 문제이다.

 

풀이:

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

A = list(map(int,input().split()))
new_A = sorted(A, reverse=False)
print(new_A[K-1])

필자는 sorted 함수를 사용해서 new_A에 A를 오름차순한 값을 새로 정의하였다. reverse 파라미터를 False로 하면 오름차순, True면 내림차순으로 정렬한다. (디폴트값은 오름차순)

그 다음 K의 수에 해당하는 값을 출력하면 되는데 주의할 점은 1을 빼주어야 한다는 점이다. 파이썬에서 인덱스는 0부터 시작하기 때문에 이를 적용하기 위해서 1을 빼준 값을 출력하면 된다!