[Programmers] Lv.0 / 특별한 이차원 배열 1 / Java

문제

 

풀이

class Solution {
    public int[][] solution(int n) {
        int[][] arr = new int[n][n];
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                arr[i][j] = (i == j) ? 1 : 0;
            }
        }
        return arr;
    }
}

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

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/181833