정사각형으로 만들기

문제 설명

이차원 정수 배열 arr이 매개변수로 주어집니다. arr의 행의 수가 더 많다면 열의 수가 행의 수와 같아지도록 각 행의 끝에 0을 추가하고, 열의 수가 더 많다면 행의 수가 열의 수와 같아지도록 각 열의 끝에 0을 추가한 이차원 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

1 ≤ arr의 길이 ≤ 100
1 ≤ arr의 원소의 길이 ≤ 100
arr의 모든 원소의 길이는 같습니다.
1 ≤ arr의 원소의 원소 ≤ 1,000

나의 코드

import java.util.Arrays;

class Solution {
    public int[][] solution(int[][] arr) {
        int rows = arr.length;
        int cols = arr[0].length;
        int[][] rowsAnswer = new int[rows][rows];
        int[][] colsAnswer = new int[cols][cols];
        if(rows>cols) {
            for(int i=0; i<rows; i++) {
                rowsAnswer[i] = Arrays.copyOf(arr[i], rows);
            }
            return rowsAnswer;
        } else if(cols>rows) {
            for(int i=0; i<rows; i++) {
                colsAnswer[i] = Arrays.copyOf(arr[i], cols);
            }
            return colsAnswer;
        } else return arr;
    }
}

다른 사람 코드

class Solution {
    public int[][] solution(int[][] arr) {
        int max = Math.max(arr.length, arr[0].length);
        int[][] array = new int[max][max];

        for (int i = 0; i < arr.length; i++) {
            System.arraycopy(arr[i], 0, array[i], 0, arr[0].length);
        }
        return array;
    }
}

Math.max() 함수를 통해 하나의 결과 배열로도 풀이 가능

느낀 점

int배열은 값이 없다면 자동으로 0으로 채워지기 때문에 행이 길 때와 열이 길 때의 배열을 나눠 각각 긴 길이로 정사각형 배열을 선언해준 후, 주어진 배열 arr을 복사해주기만 하면 된다. 여기서, 이차원 배열이기 때문에 arr[i]를 복사하는 것이므로 주의해야한다.
다른 사람 코드처럼 Math.max()함수를 활용하여 굳이 행이 길 때와 열이 길 때를 나누지 않고, 하나의 변수로도 최대값을 구하여 풀이가 가능하다.