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

문제

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

제한사항

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

문제 링크

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

프로그래머스

 

풀이[c++]

#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    vector<int> temp;
    int row = arr1.size();
    int col = arr1[0].size();
    for(int i=0;i<row;i++){
        for(int j=0;j<col;j++){
            temp.push_back(arr1[i][j]+arr2[i][j]);
        }
        answer.push_back(temp);
        temp.clear();
    }
    return answer;
}

이 문제는 이중 벡터를 다루는 방법을 알면 쉽게 풀 수 있다.

row = arr1.size()로 행을 구하고, col = arr1[0].size()를 통해 열을 구한다. 그 후 temp라는 vector에 한 행의 합을 for문을 사용해 구한 후 answer에 temp를 넣는 방식을 진행한다.