[프로그래머스] 코딩테스트 연습월간 코드 챌린지 시즌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의 길이)

 

입력 & 출력

입출력 예

abresult

[1,2,3,4]	[-3,-1,0,2]	3
[-1,0,1]	[1,0,-1]	-2

새로 사용한 함수

zip() , 두 가지 변수를 엮어주는 함수입니다.

for문 활용 시에 두 변수를 같이 돌려줘야 할 때나 enumerate 등을 이용해서 인덱스 접근해서 사용하려고 할 때 대신 사용할 수 있어서 좋습니다.

 

풀이 과정

1. 두 가지 변수를 zip 함수로 엮어서 for문을 이용하여 하나씩 빼낸다.

2. 빼낸 수를 곱해서 더해주며 이것을 answer 변수에 저장한다. 마지막에 이것을 리턴해준다.

 

코드

def solution(a, b):
    answer = 0
    for pair in zip(a, b):
        answer += pair[0] * pair[1]

    return answer