프로그래머스 Lv.0 특별한 이차원 배열 1

 

<Solution>

class Solution {
    public int[][] solution(int n) {
        int[][] answer = new int [n][n];
        for(int i = 0; i < answer.length; i++) {
            for(int j = 0; j < n; j++) {
                if( i == j) {
                    answer[i][j] = 1;
                }
            }
        }
        return answer;
    }
}