알고리즘/프로그래머스

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

잘하다연 2021. 1. 17. 21:46

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의 인덱스 길이를 이용해 값을 각각 곱해 덧셈연산을 더해 결과값을 구했다.

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

 

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)]