배열의 원소만큼 추가하기

문제

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

프로그래머스

 

내 답안 / 다른 사람들의 답안

import Foundation

func solution(_ arr:[Int]) -> [Int] {
    return arr.map { Array(repeating: $0, count: $0) }.flatMap { $0 }
}

- arr의 각 요소마다 Array(repeating: count:)를 사용해서 반복되는 배열로 만들고, flatMap을 통해 2차원 배열을 1차원 배열로 만들어 반환

 

func solution(_ arr: [Int]) -> [Int] {
    return arr.flatMap { Array(repeating: $0, count: $0) }
}

 

 

점수: +1