[0단계/1점]특별한 이차원 배열 1

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<answer[i].length;j++){
                if(i==j){
                    answer[i][j] = 1;
                }else{
                    answer[i][j] = 0;
                }
            }
        }
        return answer;
    }
}

 

개선 코드

 

사실 i==j가 필요 없는 이유는 배열 초기값은 0이고, 0,0 1,1 2,2와 같이 대각선이 같으면 1

        int[][] answer = new int[n][n]; // 모든 요소는 0으로 초기화됨
        for (int i = 0; i < n; i++) {
            answer[i][i] = 1; // 대각선에만 1 할당
        }
        return answer;