[프로그래머스 코딩테스트] lv1. 내적

문제 설명

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

 

나의 풀이

def solution(a, b):
    answer = 0
    for i, val in enumerate(a):
        answer += (val * b[i])
    return answer

for문을 통해서 내적 값을 구해주었다. 이 문제는 zip()함수를 통해 더 간단하게 해결할 수 있다.

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

 

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/70128