행렬의 덧셈

magicdrill·2024년 3월 8일

행렬의 덧셈

#include <string>
#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 col = arr1.size(), row = arr1[0].size();
    int i, j;
    
    for(i = 0; i < col; i++)
    {
        for(j = 0; j < row; j++)
        {
            temp.push_back(arr1[i][j] + arr2[i][j]); 
        }
        answer.push_back(temp);
        temp.erase(temp.begin(), temp.end());
    }
    
    return answer;
}

0개의 댓글