[프로그래머스] 내적(itertools 사용)

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

문제 설명

길이가 같은 두 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 이하입니다.

문제풀이

풀이보다는, 파이썬으로 코딩테스트를 볼 경우, 라이브러리를 잘 활용한다면 도움이 많이 되는데, 여기서 itertools 라이브러리의 zip 메서드가 얼마나 좋은지 알 수 있다.

zip메서드는 동시에 여러 배열에서의 값들을 가져오는 기능을 수행하는데, 이 메서드를 사용함으로써 코드의 길이가 확실히 감소하고 보기 쉬워짐을 알 수 있다.

코드

import itertools

def solution(a, b):
    answer = 0
    for _a, _b in zip(a, b):
        answer += _a*_b
    return answer