#프로그래머스 파이썬 Lv.1 #K번째수

문제설명
: 배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.
배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성

문제풀이

1. 우선 반복문을 사용해 commands 2차원 배열을 command에 1차원 배열로 따로 저장
2. arr로 문제에서 주어진 크기만큼 자르기
3. 그 다음 sort()함수를 써서 정렬
4. 그 후 k번째 수를 찾아서 answer배열에 저장 후 리턴

def solution(array, commands):
    answer = []
    
    for i in commands:
        arr = array[i[0]-1:i[1]]
        arr.sort()
        answer.append(arr[i[2]-1])  
        
    return answer

 

# 다른사람 코드

map 과 lambda 함수
- map(함수, 리스트) 여러 개의 데이터를 한번에 다른 형태로 변환하기 위해서 사용됩니다.
- lambda 인자 : 표현식

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

 

# 다른사람 코드

비슷하지만 다른데

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

 

https://programmers.co.kr/learn/courses/30/lessons/42748

코딩테스트 연습 - K번째수