[Programmers][Python][Level 1] 내적

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

 

 

 

입출력 예

a	b	result
[1,2,3,4]	[-3,-1,0,2]	3
[-1,0,1]	[1,0,-1]	-2

 

입출력 예 설명

입출력 예 #1

• a와 b의 내적은 1*(-3) + 2*(-1) + 3*0 + 4*2 = 3 입니다.

 

입출력 예 #2

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

풀이

다음과 같이 문제를 접근했습니다.

 

1. a와 b의 내적은 a[i+0] * b[i+0] + ... a[i+n]  * b[i + n]을 구하면 되기에 a의 갯수만큼 for문을 돌려줍니다.
2. answer 변수에 a * b를 곱한 값을 복합 연산자를 통해 더해주면 됩니다.

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

 

 

zip() 사용한 풀이

• zip() : 파라미터에 있는 리스트들의 값을 병렬적으로 받아서 처리가 가능합니다.

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