Basic56 리스트 자르기

문제: 리스트 자르기

문제 설명

정수 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 함수를 완성해주세요.

---

제한사항

• n 은 1, 2, 3, 4 중 하나입니다.
• slicer의 길이 = 3

• slicer에 담긴 정수를 차례대로 a, b, c라고 할 때

  • 0 ≤ a ≤ b ≤ num_list의 길이 - 1 
  • 1 ≤ c ≤ 3

• 5 ≤ num_list의 길이 ≤ 30

• 0 ≤ num_list의 원소 ≤ 100

---

입출력 예

n	slicer	num_list	result
3	[1, 5, 2]	[1, 2, 3, 4, 5, 6, 7, 8, 9]	[2, 3, 4, 5, 6]
4	[1, 5, 2]	[1, 2, 3, 4, 5, 6, 7, 8, 9]	[2, 4, 6]

---

입출력 예 설명

입출력 예 #1

• [1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1번 인덱스부터 5번 인덱스까지 자른 리스트는 [2, 3, 4, 5, 6]입니다.

입출력 예 #2

• [1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1번 인덱스부터 5번 인덱스까지 2개 간격으로 자른 리스트는 [2, 4, 6]입니다.

---

Solution

switch case로 n의 4가지 경우를 구분하고, 각각 조건에 맞게 구현했습니다. n이 4일 때 c 간격으로를 구현하는 방법은 여러가지가 있겠지만, %연산자를 이용해서 구했습니다.

import Foundation

func solution(_ n:Int, _ slicer:[Int], _ num_list:[Int]) -> [Int] {
    let (a, b, c) = (slicer[0], slicer[1], slicer[2])
    switch n {
        case 1:
            return num_list[0...b].map{$0}
        case 2:
            return num_list[a...].map{$0}
        case 3:
            return num_list[a...b].map{$0}
        case 4:
            return num_list[a...b].indices.filter{$0%c == 0}.map{num_list[$0]}
        default:
            return [0]
    }
}