[Lv.0_Java] 특별한 이차원 배열 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.length; j++){
                if(i==j){
                    answer[i][j] = 1;
                }else{
                    answer[i][j] = 0;
                }
            }
        }
        return answer;
    }
}

nxn 배열이기에 answer 배열의 길이를 설정해주고 이중 for 문을 돌려서 i == j 일 경우 1 그 위치에 1 을 넣어주고 아닐경우 0 을 넣어주라고 코드를 작성하였다.