본문 바로가기

CodingTest/Level1

[연습]코딩테스트-k번째수

코딩테스트 연습 문제

  • programmers.co.kr
  • k번째수 문제

 

def solution(array, commands):
    answer = []
    for x in commands:
        t1 = array[x[0]-1:x[1]]
        t1.sort()
        answer+=[t1[x[2]-1]]
    return answer


그냥 슬라이싱 하는 문제였음.



다른사람 풀이보니 lambda써서 한줄코딩하더라,

def solution(array, commands):
    return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))
반응형