[프로그래머스] 181897번 : 리스트 자르기 (Swift)

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

프로그래머스

---

문제 설명

정수 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

 

입출력 예

 

 

입출력 예 설명

 

 

풀이

 

첫 번째 시도

import Foundation

func solution(_ n:Int, _ slicer:[Int], _ num_list:[Int]) -> [Int] {
    var firstIndex = 0
    var lastIndex = 0
    
    switch n {
    case 1:
        firstIndex = 0
        lastIndex = slicer[1]
        return Array(num_list[firstIndex...lastIndex])
    case 2:
        firstIndex = slicer[0]
        lastIndex = num_list.count - 1
        return Array(num_list[firstIndex...lastIndex])
    case 3:
        firstIndex = slicer[0]
        lastIndex = slicer[1]
        return Array(num_list[firstIndex...lastIndex])
    default:
        firstIndex = slicer[0]
        lastIndex = slicer[1]
        return stride(from: firstIndex, through: lastIndex, by: slicer[2]).map{num_list[$0]}
    }
    
}

 

스위치문을 쓰다보니 같은 코드가 여러 번 반복되는 게 맘에 안듦...

 

 

 

두 번째 시도

import Foundation

func solution(_ n:Int, _ slicer:[Int], _ num_list:[Int]) -> [Int] {
    switch n {
    case 1:
        return Array(num_list[...slicer[1]])
    case 2:
        return Array(num_list[slicer[0]...])
    case 3:
        return Array(num_list[slicer[0]...slicer[1]])
    default:
        return stride(from: slicer[0], through: slicer[1], by: slicer[2]).map{num_list[$0]}
    }
    
}

 

생각해보니 firstIndex와 lastIndex를 선언해주는 과정이 굳이 필요 없을 것 같아 생략했더니 더 깔끔하고 좋아보임

 

하지만 가독성이나 유지 보수 측면에서 볼 때는 1번 풀이가 더 좋은 듯하다

2번 풀이는 어디까지나 코테용