새소식

⌨️ Algorithms/프로그래머스

[Python] 프로그래머스 Lv1_내적

2022. 10. 6. 19:27

  • -

https://school.programmers.co.kr/learn/courses/30/lessons/70128

 

프로그래머스

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

programmers.co.kr

 

  • a와 b의 내적
    • a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1]

 

"""
입출력 예시)

a = [1,2,3,4], b = [-3,-1,0,2] -> 3
a = [-1,0,1], b = [1,0,-1] -> -2
"""

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

-> a와 b의 길이가 같기 때문에 for문 하나로 계산

 

 

  • 더 간단한 코드
def solution(a, b):
    return sum([a[i] * b[i] for i in range(len(a))])

-> 리스트를 sum해서 구함

Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다!