💡（Java）특별한 이차원 배열 1

• 문제 설명
  
  정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를 return 하는 solution 함수를 작성해 주세요.
  
  arr[i][j] (0 ≤ i, j < n)의 값은 i = j라면 1, 아니라면 0입니다.

 

• 제한사항
  
  1 ≤ n ≤ 100
  
  

 

• 입출력 예
  
  

  

  
  
  
  
• 입출력 예 설명
  
  

  

  

---

🌀

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

 

[풀이 포인트]

→ 배열의 기본 초기화 값

• Java에서 정수형 배열 생성 시, 모든 원소는 자동 0으로 초기화
  → 따로 0을 채워넣을 필요 없다.
• 조건에서 i와 j가 같은 위치 (주대각선)만 1로 설정하면 된다.
  → for문을 통해 각 행의 i번째 원소(즉, arr[i][i])에 1을 할당.