알고리즘/프로그래머스
프로그래머스 Lv1 K번째수(python)
개발하는 호랑이
2024. 1. 29. 19:52
K번째수
정렬할 줄 아냐는 문제다.
문제만 잘 읽으면 python을 쓸 줄 아는 사람이라면 바로 풀이 가능하다.
여기서 i와 j, k는 컴퓨터에서 계산하는 0부터 시작이 아닌 1부터 시작이니 그 부분만 주의하면 된다.
풀이
def solution(array, commands):
answer = []
for a in range(len(commands)):
i = commands[a][0]
j = commands[a][1]
k = commands[a][2]
answer.append(sorted(array[i-1:j])[k-1])
return answer
다른 사람은 아래처럼 풀었다.
def solution(array, commands):
return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))