문제

내 최종코드

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++){
                answer[i][j] = i == j ? 1 : 0;
            }
        }
        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;
    }
}
  • i와 j가 같을 경우에만 1을 넣어주면 되기 때문에 위와 같은 코드로 작성하면 for문을 한 번만 작성해도 된다.