프로그래머스 lv1 내적(월간 코드 챌린지 시즌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 이하입니다.
  

입출력 예

입출력 예 설명

입출력 예 #1

a와 b의 내적은 1(-3) + 2(-1) + 30 + 42 = 3 입니다.

입출력 예 #2

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

접근

1. 같은 순서의 인덱스를 곱한고

2. 다 더 해주면 됨

풀이

def solution(a, b):
    adotb = 0
    for i in range(len(a)):
        adotb += a[i]*b[i]
    return adotb

if __name__ == "__main__":
    a, b = [1,2,3,4], [-3,-1,0,2] # result = 3
    print(solution(a,b))

# 다른 사람 풀이.  리스트 컴프리헨션
def solution(a, b):

    return sum([x*y for x, y in zip(a,b)])

if __name__ == "__main__":
    a, b = [1,2,3,4], [-3,-1,0,2] # result = 3
    print(solution(a,b))

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges