[프로그래머스] Lv.1 행렬의 덧셈 c++

https://school.programmers.co.kr/learn/courses/30/lessons/12950

프로그래머스

 

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 된다.

 

2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환해라.

 

arr1 = [[1, 2], [3, 4]]  

arr2 = [[5, 6], [7, 8]]

 

answer = [[6,8,], [10,12]]

 

풀이 방법

2차원 백터 배열 입력방법만 잘 알고 있으면 된다.

임시 배열 row를 할당하여 각 행렬의 원소를 덧셈한걸 넣어준다.

그리고 그 row배열을 answer 백터에 추가해주면 된다.

풀이 코드

#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> row;
        for(int j=0; j<arr1[i].size(); j++){
            row.push_back(arr1[i][j] + arr2[i][j]);
        }
        answer.push_back(row);
    }
    return answer;
}