본문 바로가기
Programmers Coding-Test

[Python] 정렬 : k번째수

by 으녜 2021. 4. 4.
728x90

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

 

💡문제💡


배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하시오.

 

[제한사항]

  • array의 길이는 1 이상 100 이하입니다.
  • array의 각 원소는 1 이상 100 이하입니다.
  • commands의 길이는 1 이상 50 이하입니다.
  • commands의 각 원소는 길이가 3입니다.

[입출력 예시]

 

 

코드 및 풀이💻

 


 

1 ) commands의 길이만큼 반복

2 ) commands의 [i][0]번째 요소와 commands의 [i][1]번째 요소 범위만큼 슬라이싱 후 sorted(정렬)함

 => 이때, array_slice의 index는 0으로 시작하므로, 해당 번째부터 뽑아내기 위해서는 -1를 해줘야함

3 ) commands[i][2]번째에 해당하는 요소를 answer 리스트에 추가

728x90