[프로그래머스] 행렬의 덧셈 (C++)

행렬의 덧셈 / Lv.1

---

문제  설명 )

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 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) {
    vector<vector<int>> answer;
    
    // answer vector의 크기를 미리 키워놓는다.
    answer.resize(arr1.size());
    
    // 해당 행렬의 원소들을 더해 answer에 넣어준다.
    for(int i = 0; i < arr1.size(); i++)
    {
        for(int j = 0; j < arr1[i].size(); j++)
        {
            answer[i].push_back(arr1[i][j] + arr2[i][j]);
        }
    }
    
    
    return answer;
}

---

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

프로그래머스