[코딩테스트 Lv.0] 정사각형으로 만들기

문제 설명

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

제한사항

1 ≤ arr의 길이 ≤ 100
1 ≤ arr의 원소의 길이 ≤ 100
    arr의 모든 원소의 길이는 같습니다.
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]]

입출력 설명

입출력 예 #1
예제 1번의 arr은 행의 수가 4, 열의 수가 3입니다. 행의 수가 더 많으므로 열의 수를 4로 만들기 위해 arr의 각 행의 끝에 0을 추가한 이차원 배열 [[572, 22, 37, 0], [287, 726, 384, 0], [85, 137, 292, 0], [487, 13, 876, 0]]를 return 합니다.

입출력 예 #2
예제 2번의 arr은 행의 수가 2, 열의 수가 4입니다. 열의 수가 더 많으므로 행의 수를 4로 만들기 위해 arr의 각 열의 끝에 0을 추가한 이차원 배열 [[57, 192, 534, 2], [9, 345, 192, 999], [0, 0, 0, 0], [0, 0, 0, 0]]을 return 합니다.

입출력 예 #3
예제 3번의 arr은 행의 수와 열의 수가 2로 같습니다. 따라서 0을 추가하지 않고 [[1, 2], [3, 4]]을 return 합니다.

---

내 풀이

Java

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

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// arr_rows는 2차원 배열 arr의 행 길이, arr_cols는 2차원 배열 arr의 열 길이입니다.
int** solution(int** arr, size_t arr_rows, size_t arr_cols) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    int len = arr_rows > arr_cols ? arr_rows : arr_cols;
    int** answer = (int**)malloc(sizeof(int*)*len);
    
    for(int i = 0; i < len; i++){
        answer[i] = (int*)malloc(sizeof(int)*len);
        if(i < arr_rows){
            for(int j = 0; j < len; j++){
                if(j < arr_cols){
                    answer[i][j] = arr[i][j];
                }else{
                    answer[i][j] = 0;
                }
            }
        }else{
            for(int j = 0; j < len; j++){
                answer[i][j] = 0;
            }
        }
    }
    
    return answer;
}

Python

def solution(arr):    
    ms = max(len(arr), len(arr[0]))
    
    answer = [[0]*ms for _ in range(ms)]
    for i in range(len(arr)):
        for j in range(len(arr[i])):
            answer[i][j] = arr[i][j]
    
    return answer

---

https://school.programmers.co.kr/learn/courses/30/lessons/181830

프로그래머스