행렬의 덧셈

김재혁·2025년 4월 30일

문제

  • 행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 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에 저장

0개의 댓글