프로그래머스 - java 특별한 이차원 배열 1

체감 Level : ★☆☆  Review: 2차원 배열 행,열 이해문제!

 

 

💡 nxn 배열에서 순서대로 하나씩 1을 찍어주는 배열을 만들어야한다.

 

보통 2차원 배열이라고해서 2중 for 문을 기본으로 해야겠다 ! 싶겠지만

이 문제는 하나의 for문으로 풀 수 있다.

 

1) 0번째 행 0번을 1로 찍어주고

2) 1번째 행 1번을 1로 찍어주고

3) 2번째 행 2번을 1로 만들어준다.

 

즉 [i][i] 로 풀 수 있는 문제 !!

 

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