[Programmers] #181833 - 특별한 이차원 배열 1 [Java][C++][Python]

문제 링크

---

1. 아이디어

이차원 배열을 $i = j$면 1 아니면 0으로 채운 후 반환하는 문제다. 모든 값이 0으로 초기화된 이차원 배열을 만든 후 행 인덱스와 열 인덱스가 일치할 때만 1로 바꿔서 반환했다.

---

2. 복잡도

접근	시간	공간
풀이	$O(N^2)$	$O(N^2)$

($N$ = 입력값 n)

---

3. 코드

풀이 [Java][C++][Python]

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

        return arr;
    }
}

#include <bits/stdc++.h>
using namespace std;

vector<vector<int>> solution(int n) {
    vector<vector<int>> arr(n, vector<int>(n));
    for (int i = 0; i < n; i++) {
        arr[i][i] = 1;
    }

    return arr;
}

def solution(n):
    arr = [[0] * n for _ in range(n)]
    for i in range(n):
        arr[i][i] = 1

    return arr

---