본문 바로가기
코딩테스트 연습

정사각형으로 만들기

by 신재은👩🏼‍💻 2024. 4. 25.

문제 설명

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

 

제한사항

  • 1 ≤ arr의 길이 ≤ 100
  • 1 ≤ arr의 원소의 길이 ≤ 100
  • 1 ≤ arr의 원소의 원소 ≤ 1,000

 

입출력 예

arr result
[[572, 22, 37], [287, 726, 384], [85, 137, 292], [487, 13, 876]] [[572, 22, 37, 0], [287, 726, 384, 0], [85, 137, 292, 0], [487, 13, 876, 0]]
[[57, 192, 534, 2], [9, 345, 192, 999]] [[57, 192, 534, 2], [9, 345, 192, 999], [0, 0, 0, 0], [0, 0, 0, 0]]
[[1, 2], [3, 4]] [[1, 2], [3, 4]]

class Solution {
    public int[][] solution(int[][] arr) {
        int row = arr.length;
        int column = arr[0].length;

        if (row > column) {
            int[][] newArr = new int[row][row];
            for (int i = 0; i < row; i++) {
                for (int j = 0; j < row; j++) {
                    if (j < column) {
                        newArr[i][j] = arr[i][j];
                    } else {
                        newArr[i][j] = 0;
                    }
                }
            }
            return newArr;
        } else if (column > row) {
            int[][] newArr = new int[column][column];
            for (int i = 0; i < column; i++) {
                for (int j = 0; j < column; j++) {
                    if (i < row) {
                        newArr[i][j] = arr[i][j];
                    } else {
                        newArr[i][j] = 0;
                    }
                }
            }
            return newArr;
        } 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;

    }

}

 

이 코드를 읽자 마자 내 입에서 '미친...'소리가 절로 나왔다.

모르거나 안 써 본 메서드가 하나도 없다.

근데 진짜 이렇게 할 생각을 0.1g도 해 내지 못 했다.

 

와, 진짜 쉽고, 간단하고, 단순하고, 좋은 코드다...

 

나는 도대체 뭐가 문젤까?

좀 더 '입체적인' 사고를 못 하는 건가???

'코딩테스트 연습' 카테고리의 다른 글

x만큼 간격이 있는 n개의 숫자 풀이 - Lv. 1  (0) 2024.04.25
하샤드 수 풀이 - 프로그래머스 Lv. 1  (0) 2024.04.25
소인수분해  (0) 2024.04.24
문자 개수 세기  (0) 2024.04.24
정수 제곱근 판별  (0) 2024.04.24