본문 바로가기
알고리즘/백준 solved.ac PS

[백준][solved.ac][Silver 5][Python] 11004번 : K번째 수

by 빛밤하늘 2021. 5. 26.
반응형

밑의 링크는 백준에서의 문제 링크입니다.

 

11004번: K번째 수

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

www.acmicpc.net

 

 

 

 

 

 

 

 

파이썬 코드와 결과입니다.

# readline을 사용하기 위해 import합니다.
from sys import stdin

# 첫째 줄에 N, K를 공백으로 구분해 입력합니다.
# 1 <= N <= 5,000,000
# 1 <= K <= N
# 각각 정수형으로 변환합니다.
N, K = map(int, stdin.readline().split(' '))
# 둘째 줄에는 A1, A2, ..., AN을 공백으로 구분해 입력합니다.
# -10^9 <= Ai <= 10^9
# 각각 정수형으로 변환하고 리스트 변수에 넣은 뒤, 오름차순으로 정렬합니다.
sorted_numbers = sorted(list(map(int, stdin.readline().split(' '))))

# 앞에서부터 K번째 있는 수를 출력합니다.
print(sorted_numbers[K - 1])

결과

 

코드를 제출하고 난 뒤, 채점이 엄청 오래 걸리길래 뭔가 잘못되었나 했는데

데이터셋이 되게 크게 설정되어 있나봅니다.

다른 사람들의 파이썬 결과들을 보니까 다들 실행 시간이 긴 편이었습니다.

 

 

 

※ 궁금한 부분, 이상한 점 및 오타는 댓글에 부탁드립니다.

※ 더 효율적이고 빠른 정답을 환영합니다.

반응형

댓글