본문 바로가기

Python 코딩테스트

정렬: 백준 11004 파이썬

반응형

문제: 

https://www.acmicpc.net/problem/11004

 

11004번: K번째 수

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

www.acmicpc.net

 

풀이:

n, k = map(int, input().split())

array = list(map(int, input().split()))

array.sort()
print(array[k-1])

 

반성:

처음에는 이 문제를 계수 정렬로 해결하려고 했었다. 하지만 이 문제의 수 범위는 0부터 5,000,000까지 였고, 

계수 정렬이 효율적일 수 있는 한계인 1,000,000을 넘어선 범위였다.

파이썬 정렬 라이브러리로 아주 쉽게 해결이 가능했다. 

반응형

'Python 코딩테스트' 카테고리의 다른 글

정렬: 백준 2108 파이썬  (0) 2021.08.29
정렬: 백준 18868 파이썬  (0) 2021.08.23
DFS/BFS: 백준 1260 파이썬  (0) 2021.08.18
DFS/BFS: 백준 2606 파이썬  (0) 2021.08.18
DFS/BFS: 백준 16173 파이썬  (0) 2021.08.12