[C++] 프로그래머스 Level 1 : 행렬의 덧셈

Kim Nahyeong·2022년 8월 1일
0

프로그래머스

목록 보기
10/38

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

for문 2개로 행렬의 곱을 구할 수 있다.
2차원 배열(벡터)로 행렬이 주어졌기 때문에 행과 열을 잘 따져서 for문을 구축해야할 필요가 있다.

arr1.size()로 행렬의 열의 크기를
arr1[0].size()로 행렬의 행의 크기를 구했다.

0개의 댓글