프로그래머스 Lv.1 코딩테스트

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

하다블 2022. 2. 9. 18:40
반응형

문제는 다음과 같습니다.

코딩테스트 연습 - 행렬의 덧셈 | 프로그래머스 (programmers.co.kr)

 

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

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

programmers.co.kr

 

지난 문제에서 벡터를 더 자유롭게 다룰 수 있는 가를 확인하는 문제라고 생각합니다.

하지만 어려울 것 없이 벡터로 되어있는 원소들의 크기를 이용하여 이중 for문을 사용하면 구할 수 있다고 생각합니다.

#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[i].size();j++)
        {
            tmp.push_back(arr1[i][j]+arr2[i][j]);
        }
        answer.push_back(tmp);
    }
    return answer;
}
반응형