NIRVANA

[level 1] 내적 본문

Coding test(Python3)/Programmers

[level 1] 내적

녜잉 2023. 7. 14. 00:55

문제

길이가 같은 두 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의 길이)

 

 

문제 풀이 접근법1

1) 리스트 a와 b의 길이 만큼 반복문을 진행하며 a[i]와 b[i]를 곱한뒤, 해당 값을 answer에 더한다. 

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

 


다른 분들 풀이

def solution(a, b):

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

아 왜 이걸 생각 못했지...? 

냅다 반복문 쓰는 버릇 고쳐야함 진짜..