배열의 원소만큼 추가하기

Lust3r 2024. 1. 30. 13:35

728x90

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

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) }
}