본문 바로가기
Python/코딩테스트

[Python] 프로그래머스 - K번째 수

by @sseyeon_ 2020. 12. 19.
반응형

[ 정답코드 ]

 

이긴한데 .. 다른분 코드 보니까 난 정말 파이썬 활용을 못하는 것 같다 ㅎㅎ

이럴거면 그냥 C 써 세여나 ^^ 

 

def solution(array, commands):
    answer = []
    tmp = []

    for i in commands:
        if i[0] != i[1]:
            tmp = array[(i[0])-1 : i[1]]
            tmp.sort()   
            answer.append(tmp[i[2]-1])
        else:
            tmp = array[i[0]-1]
            answer.append(tmp)
    return answer

 

 

 

[ map, lambda를 이용한 풀이 ]

 

둘 다 뭔지 모른다 .. 한 수 배워갑니다 .... 

lambda가 쓰이는 곳이 많댄다 공부해야지 !!!!

 

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

 

[ i, j, k  한 번에 입력하기 ]

def solution(array, commands):
    answer = []
    for command in commands:
        i,j,k = command
        answer.append(list(sorted(array[i-1:j]))[k-1])
    return answer

이런 것도 처음 봤다 !!!! 

그러고보니 두 분 다 sort가 아닌 sorted를 쓰셨네 

이것도 공부하겠읍니다 ^^ㄱ

반응형