def solution(array, commands):
answer = []
for i in commands:
temp = array[i[0]-1:i[1]]
temp.sort()
answer.append(temp[i[2]-1])
return answer
코드 설명
commands 리스트 안에서 for 반복문을 돌려 commands[i][0:2] 에 해당하는 원소들을 리스트로서 접근하여 temp 리스트에
(i ~ j 번째를 자른 결과) 넣어준다. 그 후 정렬을 하고 정렬된 리스트의 i[2]-1 인덱스(k를 의미함)에 접근하여 answer 리스트에 원소를 추가하여 준다.
Point : 파이썬에는 내장 정렬 함수인 sort(), sorted()가 존재하기에 이를 활용하면 정렬을 위한 코드를 작성하지 않아도 된다. 또한 리스트 안에서 for 반복문을 사용할 수 있으므로 보다 리스트에 쉽게 접근하여 사용할 수 있다.
728x90
'SW개발 > 코딩테스트' 카테고리의 다른 글
[프로그래머스]H-Index-정렬 (0) | 2020.11.26 |
---|---|
[프로그래머스]가장 큰 수-정렬 (0) | 2020.11.25 |
[프로그래머스]네트워크-DFS/BFS (0) | 2020.11.20 |
[프로그래머스]타겟넘버-DFS/BFS (0) | 2020.11.19 |
[프로그래머스]체육복-그리디 (0) | 2020.10.30 |