[Swift Algorithm] 내적 Programmers(Lv.1)

문제 설명

길이가 같은 두 1차원 정수 배열 a, b가 매개변수로 주어집니다. a와 b의 내적을 return 하도록 solution 함수를 완성해주세요.

이때, a와 b의 내적은 a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1] 입니다. (n은 a, b의 길이)

 

제한 조건

• a, b의 길이는 1 이상 1,000 이하입니다.
• a, b의 모든 수는 -1,000 이상 1,000 이하입니다.

 

입출력 예

a	b	result
[1,2,3,4]	[-3,-1,0,2]	3
[-1,0,1]	[1,0,-1]	-2

 

내 문제 풀이

import Foundation

func solution(_ a:[Int], _ b:[Int]) -> Int {
    var sum = 0
    for i in 0..<a.count {
        sum += a[i] * b[i]
    }
    return sum
}

• 두 배열의 길이가 같으므로, 그냥 a 배열 길이까지 반복문을 돌려줬고, 두 배열의 같은 원소 값을 곱해서 누적했다.

 

인상적인 풀이

import Foundation

func solution(_ a:[Int], _ b:[Int]) -> Int {
    return zip(a, b).map(*).reduce(0, +)
}

• zip을 사용하여 a, b 배열의 같은 위치 요소들을 곱해준 후 모두 더해주었다.!

 

💡 피드백

• zip은 거의 사용해본 적이 없어서 생각지도 못했는데, 이렇게 활용할 수 있다고 배울 수 있었다,,!!!

 

---

 

문제

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