여유로운 금요일 쉬운 알고리즘 문제를 풀어보자. 레벨 1 문제다 ㅎ.ㅎ 오랜만에 프로그래머스 들어가니 2000등으로 밀렸군.

내적

문제 설명

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

작성한 코드

Javascript

function solution(a, b) {
    return a.reduce((prev, val, idx) => prev + val * b[idx], 0)
}

Python

def solution(a, b):
    result = list(map(lambda x,y : x*y, a,b))
    totalSum = sum(result)
    return totalSum

설명

와 요즘 JS만 쓰니까 진짜 문제보자마자 JS는 알겠는데 파이썬이 생각이 안난다. 한줄만에 풀어버릴 수 있는 문제다. reduce 만세~

JS 문법을 생각해보다가 python의 람다 함수로 문제를 풀었다. map(param1, param2, ...) 일 때 param1은 함수고 이후 파라미터는 함수에 사용하는 인자들이다. 여기에 익명함수인 람다함수를 첫 번재 파라미터로 넣고 배열을 이용해서 곱한 값들을 리스트에 넣는다. 이후 리스트를 sum() 메소드를 이용해서 더한 후 리턴한다.