특별한 2차원 배열

정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를 return 하는 solution 함수를 작성해 주세요.

 

 

package com.example.demo;
class Solution {
    public int[][] solution(int n) {
        int[][] answer = new int[n][n];
        for(int i=0; i<n; i++){
            for(int j=0;j<n;j++){
                if(i==j){
                    answer[i][j]=1;
                }else {
                    answer[i][j]=0;
                }
            }

        }


        return answer;
    }
}

이번 문제는 이차원 배열에서 일정한 규칙을 가진거여서  i = j 일때만 1을 넣어주고 나머지에는 0을 넣어주면

간단히 해결이 가능하다. 

끝!!