URL : https://programmers.co.kr/learn/courses/30/lessons/70128?language=python3
문제 설명
- 길이가 같은 두 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 이하입니다.
문제풀이
- 백터의 내적을 구하는 문제
- 각 리스트들의 같은 위치를 곱한 뒤 다 더하는 코드
- for문을 사용했음
1
2
a = [1, 2, 3, 4]
b = [-3, -1, 0, 2]
1
2
3
4
5
6
def solution(a, b):
answer = 0
for i in range(len(a)):
num = a[i] * b[i]
answer += num
return answer
1
solution(a, b)
1
3
1
2
def solution(a, b):
return sum([x * y for x, y in zip(a,b)])
1
solution(a, b)
1
3