알고리즘

[프로그래머스 lv 1] 내적.py

파뱁 2022. 1. 11. 13:06
728x90

<문제 설명>

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

 

코딩테스트 연습 - 내적

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

programmers.co.kr

제시된 두 배열(a,b)의 값을 내적하면 되는 문제이다.

 

입출력의 예는 다음과 같다.

a b result
[1,2,3,4] [-3,-1,0,2] 3
[-1,0,1] [1,0,-1] -2

 

<문제 해결 아이디어>

문제에서 두 배열의 길이는 같다고 제시했기 때문에,

같은 반복문 안에서 해결할 수 있다.

배열의 길이 만큼 반복문을 돌면 되고, 반복문 안에서는 a[i] * b[i]를 수행하여 answer에 더해주면 되는 문제이다.

 

<문제 풀이 코드>

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

채점 결과

ps. 너무 쉬워서 업로드 할 만한 건지 잠깐 의문이 들었으나 일단 풀었으니 업로드 하였다.

728x90
반응형