[프로그래머스][Level1] 행렬의 덧셈 c++

문제 링크 https://programmers.co.kr/learn/courses/30/lessons/12950

문제

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

제한 조건

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

풀이 과정

이 문제는 프로그래머스에 익숙해야만 풀 수 있는 문제이다.
return값이 이중 벡터이기 때문에 벡터를 만들고 벡터를 넣어주는 식으로 풀어야한다.

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;

    for(int i = 0; i < arr1.size(); i++){
        vector<int> v;
        for(int j = 0; j < arr1[i].size(); j++){
            int sum = arr1[i][j] + arr2[i][j];
            v.push_back(sum);
        }
        answer.push_back(v);
    }

    return answer;
}