문제
- 행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.
풀이
#include <string>
#include <vector>
using namespace std;
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2)
{
vector<vector<int>> answer(arr1.size(), vector<int>(arr1[0].size()));
for(int i = 0; i < arr1.size(); ++i)
{
for(int j = 0; j < arr1[i].size(); ++j)
{
answer[i][j] = arr1[i][j] + arr2[i][j];
}
}
return answer;
}
풀이 방식
- 출력용 행열(answer) 생성 : arr1이나 arr2와 행열의 갯수가 같아야 하므로 arr1의 행렬의 개수를 기준으로 answer백터를 초기화
- 이중반복문을 사용하여 순서대로 행(i), 열(j)을 순회하며 각각의 합을 answer에 저장