K번째수

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42748

처음엔 result에 append 하는 부분을 잘못 잡아 에러가 났지만 list범위를 다시 생각해보고 추가하니 성공할 수 있었다.

for문을 여러번 돌려 시간 초과가 될 수 있으니 다른 방법을 찾아봐야겠다..!

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

 

아래 링크에 여기서 사용되는 sort()와 헷갈려하는 sorted()에 대한 설명을 포스팅해놨으니 참조해주세요 :)

2020/05/03 - [개발이야기/python] - sorted()와 sort()의 차이

sorted()와 sort()의 차이