문제풀이/프로그래머스
[프로그래머스] 행렬의 덧셈
Hyeon-Uk
2021. 6. 19. 21:45
반응형
https://programmers.co.kr/learn/courses/30/lessons/12950
코딩테스트 연습 - 행렬의 덧셈
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요
programmers.co.kr
-풀이-
이중 for문을 이용하여, 해당 인덱스에 위치한 각각의 원소를 서로 더해주면 된다.
-시간복잡도-
2중 for문이므로, 행렬의 길이가 NxN이라면, O(N2)이 된다.
-코드-
#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> temp;
for(int j=0;j<arr1[i].size();j++){
temp.push_back(arr1[i][j]+arr2[i][j]);
}
answer.push_back(temp);
}
return answer;
}
반응형