본문 바로가기
알고리즘/프로그래머스

프로그래머스 Lv1 K번째수(python)

by 개발하는 호랑이 2024. 1. 29.

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))