[프로그래머스] 내적

1. 문제 설명

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

2. 풀이 코드

#include <string>
#include <vector>

using namespace std;

int solution(vector<int> a, vector<int> b) {
	int answer = 0;

	for (int i = 0; i < a.size(); ++i) {

		answer += a[i] * b[i];

	}

	return answer;
}

 

3. 정리

각 배열의 길이는 같다. 배열의 길이 만큼 순회하여 각 배열의 원소의 곱을 answer 변수에 더하여 총합을 구한다.

 

전체 시간 복잡도는 배열의 원소의 갯수만큼 순회하므로  O(n) 이다.

 

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