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

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

프로그래머스

 

"""
입출력 예시)

array = [1, 5, 2, 6, 3, 7, 4], commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]] -> [5, 6, 3]
"""

def solution(array, commands):
    answer = []
    
    for i, j, k in commands:
      # array의 i번째 숫자부터 j번째 숫자까지 자르기
      first_arr = array[i-1:j] # array를 자를 때 인덱스를 0부터가 아닌 1부터로 함
      # 자른 배열을 정렬
      arr_sorted = sorted(first_arr)
      # k번째에 있는 수 구하기
      num_k = arr_sorted[k-1] # 인덱스를 0부터가 아닌 1부터로 함
      answer.append(num_k)

    return answer

 

 

 

 

참고)

 

https://velog.io/@yeonu/%ED%8C%8C%EC%9D%B4%EC%8D%AC-2%EC%B0%A8%EC%9B%90-%EB%A6%AC%EC%8A%A4%ED%8A%B8-%EC%83%9D%EC%84%B1-%EC%9A%94%EC%86%8C-%EC%A0%91%EA%B7%BC-%EB%B0%98%EB%B3%B5%EB%AC%B8%EC%9C%BC%EB%A1%9C-%EC%B6%9C%EB%A0%A5-%ED%95%A0%EB%8B%B9-%EB%B3%B5%EC%82%AC

파이썬 2차원 리스트 생성, 요소 접근, 반복문으로 출력, 할당, 복사

 

https://docs.python.org/ko/3/howto/sorting.html

정렬 HOW TO — Python 3.10.6 문서