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

김범기·2024년 1월 26일

프로그래머스

목록 보기
2/77

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))
profile
반드시 결승점을 통과하는 개발자

0개의 댓글