[Lv. 0] 정사각형으로 만들기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181830

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

이차원 정수 배열 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]]

다른 사람의 풀이 : 

import kotlin.math.max

class Solution {
    fun solution(arr: Array<IntArray>): Array<IntArray> {
        val max = max(arr.size, arr[0].size)
        val array = Array(max) { IntArray(max) }

        for (i in arr.indices) {
            System.arraycopy(arr[i], 0, array[i], 0, arr[0].size)
        }

        return array
    }
}

풀지 못해서 다른 사람의 풀이를 참고하였다.

원소(열)의 길이는 모두 같다는 것과 행, 열의 크기 비교라는 전제를 통해 max 값을 정하고

정해진 max 값으로 array의 크기를 지정해 만든 뒤 반복문 arraycopy를 통해 배열을 합친다.

 

System.arraycopy(A¹, 0², B³, 0⁴, A.size⁵)

B 배열의³ index 0에⁴ A배열의¹ index 0 부터² A배열의 사이즈⁵ 만큼 복사한다.