문제출처 : 프로그래머스

문제소개

내적
- 길이가 같은 두개의 정수배열 a,b가 매개변수로 주어짐
- a와 b의 내적을 return 하는 함수를 완성하시오

입력)
a = [1,2,3,4]
b = [-3,-1,0,2]

출력)
result = 3

코드

def solution(a, b):
    answer = 0 
    # 길이가 같은 배열의 내적은 각 요소를 곱한 뒤 더해주면 됨
    for i in range(len(a)): 
        answer += a[i]*b[i]
    return answer
a = [1,2,3,4]
b = [-3,-1,0,2]

solution(a, b)
3

리팩터링 코드

def solution(a, b):
    return sum([x*y for x,y in zip(a,b)])
profile
Data & PM

0개의 댓글