[Swift 알고리즘] - 행렬의 덧셈 (Programmers)

 

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

 

 

내 풀이

func solution(_ arr1: [[Int]], _ arr2: [[Int]]) -> [[Int]] {
  var result: [[Int]] = []

  for i in 0..<arr1.count {
    result.append([])   // 1

    for j in 0..<arr1[i].count {
      result[i].append(arr1[i][j] + arr2[i][j])  // 2
    }

  }
  return result
}

solution([[1,2], [2,3]], [[3,4], [5,6]])

 

2차원 배열 문제는 처음이라 접근하는 방법에 어려움이 있었다. 

1. 두 배열의 길이는 항상 같음으로 arr1.count를 사용해 배열안에 배열을 생성 [ [ ] ]
2. 각 배열의 요소에 접근해서 각각 더해줘야하기 때문에
   result[i번째]에 arr1[i번째][j번째] 요소 + arr2[i번째][j번째] 요소의 합 추가 
   ex)
   i가 0 //  j가 0 인경우 
       arr1[0][0]  ==> 1
       arr2[0][0] ==> 3
       result[0] 에 4추가 ==> [ [4, ] ]
   
   i가 0 // j가 1인 경우
       arr1[0][1]  ==> 2
       arr2[0][1] ==> 4
       result[0] 에 6추가 ==> [ [4, 6] ]

 

다른사람의 풀이

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

solution([[1,2], [2,3]], [[3,4], [5,6]])

 

zip은 여러 배열을 받아, 각 배열의 요소를 쌍으로 묶어줍니다.

 

ex) 

1. zip(arr1, arr2)
       [((1, 2), (3, 4)), ((2, 3), (5, 6))]
2. map { }
       $0 = [1, 2]
       $1 = [3, 4]
3. zip($0, $1)
       [(1, 3), (2, 4)]
4. map { $0 + $1 }
       1 + 3 = 4
       2 + 4 = 6
       [4, 6]

 

어우..  2차원 배열이라 헷갈리긴 한데 zip을 잘 이용하면 각 배열의 계산값을 구하는데 유용하게 사용될것 같습니다.

 

배운것

zip(_:_:)

 

예제

let nums = [0, 1, 2, 3]
let numbers = 1...4


for (num, number) in zip(nums, numbers) {
    print("\(num): \(number)")
}

// 0: 1
// 1: 2
// 2: 3
// 3: 4

 

태그 

Swift (Programmers or LeetCode), 제목, 사용메서드,