(JAVA) 행렬 덧셈의 결과를 반환하는 함수 만들기

▶ 문제

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 
서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 
행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건
행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

▶ 소스코드

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    // arr1과 arr2의 크기가 같기 때문에 arr1의 크기를 기준으로 새로운 벡터를 생성
    vector<vector<int>> answer(arr1.size(), vector<int>(arr1[0].size()));

    // 이중 for문을 사용해 각 행과 열의 원소를 더한 값을 answer에 저장
    for (int i = 0; i < arr1.size(); i++) {
        for (int j = 0; j < arr1[0].size(); j++) {
            answer[i][j] = arr1[i][j] + arr2[i][j];
        }
    }

    return answer;
}

▶ 설명

• answer 벡터는 arr1과 동일한 크기의 2차원 벡터로 생성합니다.
• 이중 for문을 사용해 arr1과 arr2의 각 원소를 더한 결과를 answer에 저장합니다.
• 각 행(i)과 열(j)을 순차적으로 순회하며 덧셈을 수행합니다.