Posts 내적 [Python]
Post
Cancel

내적 [Python]

URL : https://programmers.co.kr/learn/courses/30/lessons/70128?language=python3

문제 설명

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

제한사항

  • a, b의 길이는 1 이상 1,000 이하입니다.
  • a, b의 모든 수는 -1,000 이상 1,000 이하입니다.

문제풀이

  • 백터의 내적을 구하는 문제
  • 각 리스트들의 같은 위치를 곱한 뒤 다 더하는 코드
  • for문을 사용했음
1
2
a = [1, 2, 3, 4]
b = [-3, -1, 0, 2]
1
2
3
4
5
6
def solution(a, b):
    answer = 0
    for i in range(len(a)):
        num = a[i] * b[i]
        answer += num
    return answer
1
solution(a, b)
1
3
1
2
def solution(a, b):
    return sum([x * y for x, y in zip(a,b)])
1
solution(a, b)
1
3
This post is licensed under CC BY 4.0 by the author.

3진법 뒤집기 [Python]

두 개 뽑아서 더하기 [Python]

Comments powered by Disqus.