프로그래머스 Lv.1 Swift x만큼 간격이 있는 n개의 숫자

 

문제

solution 함수는 정수 x와 자연수 n을 입력 받아, x부터 시작해 x씩 증가하는 숫자를 n개 지니는 리스트를 리턴해야 한다. 제한 조건에 만족하는 함수 solution을 완성하라

 

 

제한조건

• x는 -10000000 이상, 10000000 이하인 정수입니다.
• n은 1000 이하인 자연수입니다.

 

입출력 예

x	n	answer
2	5	[2,4,6,8,10]
4	3	[4,8,12]
-4	2	[-4, -8]

 

 

 

풀이

 

처음 코드에서 몇 줄 더 줄일 수 있으면 줄이는게 나을 것 같아서 삼항 연산자를 사용하여 풀이해보았다.

 

    func solution(_ x:Int, _ n:Int) -> [Int64] {
        var answer: [Int64] = []        
        for _ in 0..<n {
            let temp = answer.endIndex
            if temp == 0 {
                answer.append(Int64(temp + x))
            } else if !answer.isEmpty {
                if let lastItem = answer.last {
                    answer.append(lastItem + Int64(x))
                }
            }
        }
        
        return answer
    }

 

func solution(_ x:Int, _ n:Int) -> [Int64] {
    var answer: [Int64] = []
    for _ in 0..<n { answer.endIndex == 0 ? answer.append(Int64(answer.endIndex + x)) : answer.append((answer.last ?? Int64(answer.endIndex)) + Int64(x)) }
    return answer
}