[Kotlin] 행렬의 덧셈 (프로그래머스 알고리즘)

 

 

 

 

문제

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

 

 

 

 

---

 

접근

 

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {

        return (arr1.size).map{it -> (arr1[i].toInt() + arr2[i].toInt())}
        
    }
}

• 위와 같이 arr1.size 만큼 map을 이용해 풀고 싶었는데 아직 사용법이 익숙치 않아 잘 안 됐다.

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> =
            Array(arr1.size) { i -> 
                arr1[i] + arr2[i]
        }
}

• 그래서 이렇게 했더니 정답과 달랐다. arr1과 arr2는 일차원 배열이 아니라, 다차원 배열이라는 걸 깨달았다.
• 위의 코드는 [[1,2],[2,3]] [[3,4],[5,6]] 입력 시 [[4,6],[7,9]]이 아니라,  [[1,2,3,4],[2,3,5,6]] 이 출력된다.

 

---

 

성공 코드

 

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> =
            Array(arr1.size) { i -> 
            IntArray(arr1[0].size) { j ->
                arr1[i][j] + arr2[i][j]
            }
        }
}

 

---

 

설명

 

1. Array(arr1.size)
   arr1.size 만큼의 Array를 만든다.
2. Array(arr1.size) { i -> ... }
   새로운 2차원 배열을 생성한다. 이 배열의 크기는 arr1의 행의 수와 동일하다.
   각각의 엘리멘트에 값을 넣어주기 위해 i -> 를 사용한다.
3. IntArray(arr1[0].size) { j -> ... }
   각 행의 원소를 계산하기 위해 새로운 행을 생성한다. 이 행의 크기는 arr1의 열의 수와 동일하다.
4. arr1[i][j] + arr2[i][j]
   arr1과 arr2의 같은 위치에 있는 원소를 더한다. 이를 통해 새로운 행렬의 해당 위치의 값을 구한다.
5. 해당 Array를 출력한다. 
6. 다만 궁금한 점은 지금 테스트 케이스가 최대 2차원 배열이니까 이렇게 풀리는데, 만약 더 배열의 차원이 높아진다면 어떻게 효율적으로 처리할 수 있는지 궁금해졌다.

 

---

 

다른 방법

 

class Solution {
    fun solution(arr1: Array<IntArray>, arr2: Array<IntArray>): Array<IntArray> {
        return arr1.mapIndexed { i, e1 -> e1.mapIndexed { j, e2 -> e2 + arr2[i][j] }.toIntArray() }.toTypedArray()
    }
}

• mapIndexed와 toTypedArray()를 이용했다.
• arr1.mapIndexed { i, e1 -> ... }
  arr1의 각 행(e1)과 해당 행의 인덱스(i)에 대해 매핑
• e1.mapIndexed { j, e2 -> ... }
  각 행(e1)의 요소(e2)와 해당 요소의 인덱스(j)에 대해 매핑
• e2 + arr2[i][j]
  현재 요소(e2)와 arr2의 동일한 위치에 있는 요소를 더하여 새로운 요소 값을 생성
• .toIntArray()
  새로운 행을 IntArray로 변환하여 현재 행의 값을 갱신
• .toTypedArray()
  변환된 행들을 다시 배열로 변환하여 최종 결과를 반환