[파이썬] 코딩테스트 연습 <리스트 자르기>

문제 설명
정수 n과 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list가 주어집니다. slicer에 담긴 정수를 차례대로 a, b, c라고 할 때, n에 따라 다음과 같이 num_list를 슬라이싱 하려고 합니다.

n = 1 : num_list의 0번 인덱스부터 b번 인덱스까지
n = 2 : num_list의 a번 인덱스부터 마지막 인덱스까지
n = 3 : num_list의 a번 인덱스부터 b번 인덱스까지
n = 4 : num_list의 a번 인덱스부터 b번 인덱스까지 c 간격으로
올바르게 슬라이싱한 리스트를 return하도록 solution 함수를 완성해주세요.

 

슬라이싱을 문제에서 원하는 대로 해줬습니다.

[a,b,c]번째는 사실 0, 1, 2번째 인덱스입니다.

 

def solution(n, slicer, num_list):
    answer = []
    if n == 1:
        answer = num_list[0:slicer[1]+1]
    elif n == 2:
        answer = num_list[slicer[0]:]
    elif n == 3:
        answer = num_list[slicer[0]:slicer[1]+1]
    else:
        answer = num_list[slicer[0]:slicer[1]+1:slicer[2]]
    return answer

 

n=4일 때 a인덱스부터 b인덱스까지 c간격으로 슬라이싱을 하기 위해서는

[a:b+1:c]로 만들어주면 됩니다. [시작 인덱스: 끝 인덱스 + 1: 원하는 간격]