프로그래머스/프로그래머스 Lv1

프로그래머스 / 연습문제 / 행렬의 덧셈

ENUM01 2020. 5. 22. 09:50

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

 

코딩테스트 연습 - 행렬의 덧셈

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

programmers.co.kr

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#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> answer2;
        for (int j = 0; j < arr1[i].size(); j++)
        {
            answer2.push_back(arr1[i][j] + arr2[i][j]);
        }
        answer.push_back(answer2);
    }
    return answer;
}

arr1과 arr2의 사이즈는 같기 때문에,

같은 행과 열을 가진 값끼리 더해주면 된다.

 

그리고 2차원 벡터에 푸쉬백을 하려면,

1차원 벡터를 만든 후 그 벡터를 푸쉬 백 해야 한다.