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
'알고리즘 > 프로그래머스' 카테고리의 다른 글
프로그래머스: 월간 코드 챌린지 시즌2 - 약수의 개수와 덧셈 (0) | 2022.08.02 |
---|---|
프로그래머스: 2019 카카오 블라인드 - 실패율 (0) | 2022.08.01 |
프로그래머스: 월간 코드 챌린지 시즌2 - 음양 더하기 (0) | 2022.07.28 |
프로그래머스: 2019 카카오 개발자 겨울 인턴십 - 크레인 인형뽑기 게임 (0) | 2022.07.28 |
프로그래머스: 숫자 문자열과 영단어 (0) | 2022.07.26 |
댓글