[Algorithm] 프로그래머스 - 특별한 이차원 배열 1 (Kotlin)

 

2차원 배열을 생성하되 i와 j가 모두 일치하는 경우, 즉 [0][0]과 [1][1] 같이 두 인덱스가 일치하는 지점의 값은 1로 채우고 아니면 0으로 채워야 한다.

코틀린에서 2차원 배열을 초기화하는 방법만 알면 풀 수 있다.

 

class Solution {
    fun solution(n: Int) = Array(n) { i -> IntArray(n) { j -> if (i == j) 1 else 0 } }
}