본문 바로가기

프로그래머스 (Swift)/기초

[Day24-5] 특별한 이자원 배열 1

문제

 

내 정답 코드

import Foundation

func solution(_ n:Int) -> [[Int]] {
    var result:[[Int]] = []
    
    for i in 0..<n {
        var temp:[Int] = []
        (0..<n).map {_ in temp.append(0) }
        temp[i] = 1
        result.append(temp)
    }
    
    return result
}

 

#

1. 일단 배열의 모든 값을 n만큼 0으로 채운다.

2. 그리고 순차적인 인덱스인 i인덱스 값을 1로 수정한다.


배운 기술