본문 바로가기
알고리즘/프로그래머스

프로그래머스: 월간 코드 챌린지 시즌1 - 내적

by 패쓰킴 2022. 7. 29.
728x90

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

문제 설명

길이가 같은 두 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

입출력 예 설명

입출력 예 #1

  • a와 b의 내적은 1*(-3) + 2*(-1) + 3*0 + 4*2 = 3 입니다.

입출력 예 #2

  • a와 b의 내적은 (-1)*1 + 0*0 + 1*(-1) = -2 입니다.

코드

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

다른 사람 풀이

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

zip 을 처음 알게됨

두개의 배열을 하나의 쌍(튜플)으로 만들어주는 함수

let words = ["one", "two", "three", "four"]
let numbers = 1...4

for (word, number) in zip(words, numbers) {
    print("\(word): \(number)")
}
// Prints "one: 1"
// Prints "two: 2
// Prints "three: 3"
// Prints "four: 4"
728x90

댓글