[Swift Algorithm] x만큼 간격이 있는 n개의 숫자 Programmers(Lv.1)

문제 설명

함수 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 arr = [Int64]()
    var k = x
    var num = 0
    
    while num < n {
        arr.append(Int64(k))
        k += x
        num += 1
    }

    return arr
}

• 입력 받은 n값 만큼, 빈 배열 arr에 k부터 x씩 누적되는 수를 append 해주었음.

 

인상적인 풀이

func solution(_ x:Int, _ n:Int) -> [Int64] {
    return Array(1...n).map { Int64($0 * x) }
}

• 리턴 문에 바로 1부터 n까지의 배열을 생성해주고, map을 이용해서 모든 요소에 (값 * x)를 해주었음.
• 배열에 먼저 1부터 n에 해당하는 수를 넣고, 그 다음 x를 곱해준다는 개념.
  나는 빈 배열에 주어진 수 부터 시작하여 x값을 누적시켜 준다는 개념으로 생각했는데,
  이 풀이는 접근 방식부터 내 풀이와 차이가 있다.

 

💡 피드백

• map 함수를 자유롭게 쓸 수 있으면 훨씬 효율적인 코딩이 가능할 것 같다.

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution3(_ x:Int, _ n:Int) -> [Int64] {
    var arr = [Int64]()
    for i in 0..<n { arr.append(Int64(x+(x*i))) }
    return  arr
}

• 로직은 첫 번째 풀었을 때와 비슷하지만, 훨씬 간결하게 작성했다.

 

---

 

문제

https://programmers.co.kr/learn/courses/30/lessons/12954