프로그래머스 - 행렬의 덧셈 - C++

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

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

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

 

설명

이 문제는 행과 열의 크기가 동일한 두 행렬의 덧셈 결과를 반환하는 문제이다. 

이 문제에서는 행렬을 이중벡터로 표현하여 arr1.size() 가 행의 크기, arr1[0].size()가 열의 크기가 된다. 

 

따라서 이중포문으로 바깥 for문에서 행의 크기만큼 돌면서 안쪽 for문에서 한 행을 기준으로

열의 같은 인덱스끼리 더해서 결과를 벡터 s에 저장한다. 그리고 안쪽 for문이 종료되면

결과를 answer에 추가한다. 그러면 결과적으로 한 행에 덧셈을 완료한 셈이다.