[코테/0레벨] 특별한 이차원 배열 1

 

 

 

 

ㅁ 최초풀이

 

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;
                }
            }
        }
        
        return answer;
    }
}

 

 

 

 

ㅁ 최종풀이

 

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;
    }
}