본문 바로가기
코딩 테스트 연습

[알고리즘] 프로그래머스 - 내적

by 코드뭉치 2023. 5. 4.

내적

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

1. 구해야 하는 것
    # a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1]
2. 1을 구하기 위해 필요한 것
    # 배열의 길이 
        # len()
    # for문

 

 

0부터 배열의 길이만큼 answer에 a[i]*b[i]를 더해줬다.

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

 

 

sum()과 리스트 컴프리헨션으로 축약한 코드

def solution(a, b):
    return sum([a[i]*b[i] for i in range(len(a))])

댓글