티스토리 뷰

728x90

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

 

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

[1, 5, 2, 6, 3, 7, 4] [[2, 5, 3], [4, 4, 1], [1, 7, 3]] [5, 6, 3]

programmers.co.kr

 

  • 해설:

어떤 수열을 i부터 j번째 원소까지 자르고 정렬했을 때 K번째에 위치하는 수를 출력하는 문제이다.

 

 

 

  • 풀이 :

python의 list slice을 사용하여 i부터 j번째 원소까지 자른 후 sorted함수로 정렬시킨다. 이후 K번째 인덱스를 리턴하면 된다.

 

 

 

 

1
2
3
4
5
def solution(array, commands):
    answer = []
    for c in commands:
        answer.append(sorted(array[c[0]-1:c[1]])[c[2]-1])
    return answer
cs
320x100
댓글
© 2022 WonSeok, All rights reserved