[Programmers] 프로그래머스 내적 lv1

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의 길이)

모든 테스트 통과 답안

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

내적은 리스트의 인덱스가 같아야 가능하다.
그래서 a의 인덱스 길이를 이용해 값을 각각 곱해 덧셈연산을 더해 결과값을 구했다.

상당히 간단한 방법이지만 zip과 sum의 함수를 사용하면 더 간단히 표현할 수 있다.

set을 이용한 풀이 (다른 사람 문제 답안)

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

zip함수를 배운적은 있으나 완벽히 내것으로 소화하고 사용하는덴 힘들었다.

zip() : 내장함수로 동일한 개수로 이루어진 자료형을 묶어줌

예를 들자면

a = [1,2,3]
b = [a,b,c] 

number_abc = list(zip(a,b))
print(number_abc)

[Output]
[(1,a),(2,b),(3,c)] 

같은 인덱스 값을 차례로 들고와 묶어주는 역할을 하는 것이 zip함수이다.

 

그래서 여기서 내적 또한 같은 위치의 인덱스를 각 리스트에서 뽑는 것에서 시작한다. 여기 for x,y in zip(a,b)를 사용해 뽑은 값들을 x,y로 추출해 곱했고 그 값들을 sum함수를 사용해 더하는 것으로 마무리했다.