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

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

프로그래머스

 

• 배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수 구하기
  • ex) array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면
    • 1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]
    • 1. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]
    • 1. 2에서 나온 배열의 3번째 숫자는 5
• commands: 배열 array, [i, j, k]를 원소로 가진 2차원 배열
• commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return

제한사항

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

 

"""
입출력 예시)

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

# 1) [1, 5, 2, 6, 3, 7, 4]를 2번째부터 5번째까지 자른 후 정렬 -> [2, 3, 5, 6]의 세 번째 숫자는 5
# 2) [1, 5, 2, 6, 3, 7, 4]를 4번째부터 4번째까지 자른 후 정렬 -> [6]의 첫 번째 숫자는 6
# 3) [1, 5, 2, 6, 3, 7, 4]를 1번째부터 7번째까지 자른 후 정렬 -> [1, 2, 3, 4, 5, 6, 7]의 세 번째 숫자는 3

# => [5, 6, 3]

 

 

## 의사코드 ##

# array의 i번째 원소의 인덱스: i-1

# for i,j,k in commands:
    # 1) 먼저, array[(i-1):j]로 슬라이싱
    # 2) 슬라이싱한 배열을 오름차순 정렬 -> sort()
    # 3) 정렬한 배열의 k번째 수의 인덱스: k-1
    #   -> k번째 수를 구해서 answer에 append

 

 

 

통과한 코드

 

def solution(array, commands):
    answer = []
    for i,j,k in commands:
        new_array = array[(i-1):j] # array의 i번째부터 j번째까지 자르기
        new_array.sort() # 정렬
        n = new_array[k-1] # k번째 수 
        answer.append(n)
    return answer

 

• for 문 한 번으로 2차원 리스트의 모든 원소 출력하기
  
  • for i, ..., n(안쪽 리스트의 원소 개수만큼의 변수) in 리스트:
    • => 리스트의 안쪽 리스트에서 원소 n개를 꺼냄

# for 문으로 i,j,k 한번에 뽑아내기
commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]]
for i,j,k in commands: # 리스트(commands)의 안쪽 리스트에서 원소 3개를 꺼냄
    print(i)
    print(j)
    print(k, '\n')
    
"""
2
5
3 

4
4
1 

1
7
3 
"""

 

• 리스트 슬라이싱
  • 리스트[추출하기 원하는 시작 인덱스: 추출하기 원하는 끝 인덱스+1]
    • -> 여기서 말하는 인덱스는 리스트의 인덱스(0부터 len(리스트)-1까지)
    • range()와 똑같음

# 슬라이싱 인덱스 확인
array = [1, 5, 2, 6, 3, 7, 4]
i = 2
j = 5
# array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]

print('array[i:j]:', array[i:j])
print('array[i-1:j-1]:', array[i-1:j-1])
print('array[i-1:j]:', array[i-1:j])

"""
array[i:j]: [2, 6, 3]
array[i-1:j-1]: [5, 2, 6]
array[i-1:j]: [5, 2, 6, 3]
"""

 

 

 

다른 풀이

 

def solution(array, commands):
    return [sorted(array[a[0]-1:a[1]])[a[2]-1] for a in commands]

-> a[0] = i, a[1] = j, a[2] = k

 

sorted(array[a[0]-1:a[1]])[a[2]-1] => sorted(array[i-1:j])[k-1]

 

 

 

cf) 이전에 작성했던 코드

 

https://monzheld.tistory.com/60

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

 

 

참고)

 

https://vision-ai.tistory.com/entry/%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EB%A6%AC%EC%8A%A4%ED%8A%B8-%EC%BD%9C%EB%A1%A0-%EC%8A%AC%EB%9D%BC%EC%9D%B4%EC%8B%B1-List-Slicing

파이썬 리스트 콜론, [ : ] 슬라이싱 ( List Slicing )