프로그래머스 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;
}
반응형