알고리즘 & 자료구조/코딩테스트
[프로그래머스]내적(C++)
인디아나쥰이
2021. 3. 3. 18:34
문제 설명
길이가 같은 두 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 이하입니다.
입출력 예
a b result
| [1,2,3,4] | [-3,-1,0,2] | 3 |
| [-1,0,1] | [1,0,-1] | -2 |
입출력 예 설명
입출력 예 #1
- a와 b의 내적은 1*(-3) + 2*(-1) + 3*0 + 4*2 = 3 입니다.
입출력 예 #2
- a와 b의 내적은 (-1)*1 + 0*0 + 1*(-1) = -2 입니다.
문제 자체에 풀이가 다 나와있으므로 패스한다.
#include <string>
#include <vector>
#include <iostream>
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;
}
int main(void)
{
vector<int> a = {1,2,3,4};
vector<int> b = {-3,-1,0,2};
cout << solution(a,b);
}728x90
반응형