[Swift Algorithm] 행렬의 덧셈 Programmers(Lv.1)

문제 설명

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

 

제한 조건

• 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

 

입출력 예

arr1	arr2	return
[[1, 2], [2, 3]]	[[3, 4], [5, 6]]	[[4, 6], [7, 9]]
[[1], [2]]	[[3], [4]]	[[4], [6]]

 

내 문제 풀이

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    var array1 = [Int]()
    var array2 = [[Int]]()
    var sum = 0
    
    for i in 0 ..< arr1.count {
        for j in 0 ..< arr1[i].count {
            sum = arr1[i][j] + arr2[i][j]
            array1.append(sum)
        }
        array2.append(array1)
        array1.removeAll()
    }
    
    return array2
}

• 같은 자리의 수를 더한 값을 저장해 줄 빈 배열 array1, 그리고 결과를 저장할 array2를 생성
• 입력 받은 첫 번째 배열의 길이를 기준으로 2중 for문 돌리기! (어차피 두 배열의 길이는 같으므로)
• array1에 같은 위치의 수를 더한 결과를 추가해주고,
  완성된 array1은 array2에 추가해준 후 removeAll로 비워준다.
  이런 식으로 array2를 모두 채워 리턴

 

인상적인 풀이

import Foundation

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    return zip(arr1, arr2).map{zip($0,$1).map{$0+$1}}
}

• zip 함수를 이용, 쌍으로 묶어주어 계산했다.
• 내 풀이는 내가 풀면서도 좀 복잡하다고 생각했는데,
  zip과 map을 잘 사용하니 이렇게 한 줄로도 가능하다.

 

💡 피드백

• zip 함수는 오늘 인상적인 풀이를 통해 처음 배우게 됐는데,
  두 개의 시퀀스를 쌍으로 묶어주는 것을 말한다고 한다.. 
  이건 따로 공부가 필요할 듯 하다.
• map 함수는 정말 다양하게 쓰이는 듯 하다. 자유자재로 쓸 수 있도록 연습이 필요하겠다.

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    var result = arr1
    
    for i in 0..<arr1.count {
        for j in 0..<arr1[1].count {
            result[i][j] += arr2[i][j]
        }
    }
    return result
}

• 첫 번째 풀이보다 훨씬 간결하고 쓸데 없는 코드 없이 잘 푼 것 같다.
• 인상적인 풀이 확인해보니 zip, map 활용하면 한 줄에 끝낼 수 있구나; 

 

---

 

문제

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