[프로그래머스/c++] 행렬의 덧셈(2차원 vector)

2차원 vector 사용

 

2차원 vector의 선언은 vector안에 vector를 담는다고 생각

 

vector<vector<int>> answer;

 

그리고 1차원 vector 배열을 먼저 만들고

이 1차원 vector 배열을 2차원 vector 배열에 넣어준다.

 

    for(i=0; i<rc; i++){
        vector<int> v; // 1차원 배열 만들기
        for(k=0; k<cc; k++){
            v.push_back(arr1[i][k]+arr2[i][k]); // 1차원 배열에 값 push_back
        }
            answer.push_back(v); // 2차원 배열에 1차원 배열 push_back
    }

 

---

 

행렬의 덧셈 함수 코드

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    int rc=arr1.size(), cc=arr1[0].size(), i, k; // 행의 크기, 열의 크기 구하기
    for(i=0; i<rc; i++){
        vector<int> v;
        for(k=0; k<cc; k++){
            v.push_back(arr1[i][k]+arr2[i][k]);
        }
        answer.push_back(v);
    }
    
    return answer;
}