Programmers Coding Quiz #11 내적계산

김기욱·2021년 2월 2일
0

코딩테스트

목록 보기
11/68

문제 설명

길이가 같은 두 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 이하입니다.

입출력 예

abreturn
[1,2,3,4][-3,-1,0,2]3
[-1,0,1][1,0,-1]-2

풀이

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

내적의 개념은 뭔지 모르겠지만 내적의 계산식을 문제에서 제시해줍니다.
서로 다른 리스트의 같은 인덱스에 있는 값들을 곱해주면 되네요.
같은 길이라는 가정 하에 zip을 써서 간단히 해결가능합니다.

profile
어려운 것은 없다, 다만 아직 익숙치않을뿐이다.

0개의 댓글