문제풀이/프로그래머스

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

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;
}
반응형