백준 알고리즘 | 11004 : K번째 수 (Python / 파이썬)

---

K번째 수 성공분류

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
2 초	512 MB	30810	9624	5946	36.559%

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

11004번: K번째 수

---

문제

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

​

입력

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

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

​

출력

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

알고리즘 분류

정렬

---

백준 알고리즘 # 11004번 : K번째 수

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

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

result = sorted(a)

print(result[k - 1])

풀이

n, k 변수로 각각 정수를 입력받고, a를 리스트로 초기화 한 뒤 n개의 정수를 입력받습니다.

다음으로 정렬한 값의 k번째 수를 출력해야 하니

​

result 변수를 만들고 sorted() 함수를 이용해서 정렬해줍니다.

​

key point

마지막으로 앞에서부터 K번째에 있는 수를 구해야하는데

인덱스를 그대로 이용할 경우 K로 주어진 2의 값은 실제로 3번째에 있는 값을 가리키게 되므로 [0, 1, 2,]

k - 1을 인덱스로 지정해서 출력해주면 문제 풀이 끄--읏!👏🏻

​

자세한 코드가 궁금하신 분들은 아래 GitHub 코드를 참고해주세요🙏🏻

wook2124/Algorithm-Test

최종 소스코드