[Programmers] 내적 (월간 코드 챌린지 시즌1 - Level1, Python3)

 

코딩테스트 연습 - 내적

 

문제 설명

 

길이가 같은 두 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~3분내로 풀이가 가능한 문제였다.

 

이런 문제는 조건을 잘 확인해봐야하는데 첫 번째로 봐야 할 것은 두 배열의 길이이다.

문제 처음에 "길이가 같은" 두 배열이라고 했다.

이 말은 두 배열 중 하나의 길이만 알면 반복문을 통해 두 배열의 각 인덱스에 접근이 가능하다는 것이다.

 

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