본문 바로가기
연습장

연습문제 > 행렬의 덧셈

by anothel 2021. 11. 27.

문제 설명

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

제한 조건
  • 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.
입출력 예
arr1 arr2 return
[[1,2],[2,3]] [[3,4],[5,6]] [[4,6],[7,9]]
[[1],[2]] [[3],[4]] [[4],[6]]

 

Solution

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1,
                             vector<vector<int>> arr2) {
  vector<int> v;
  vector<vector<int>> answer;

  for (int i = 0; i < arr1.size(); i++) {
    v.clear();
    for (int j = 0; j < arr1[i].size(); j++) {
      v.push_back(arr1[i][j] + arr2[i][j]);
    }
    answer.push_back(v);
  }

  return answer;
}

 

후기

이중 for문을 사용하긴 했지만 나의 Solution으로 충분해 보인다. 옆에 여자 친구(곧 와이프)가 카트라이더를 한다고 다운로드하고 있다. 할 일이 참 많다. 코딩 테스트 문제 풀기, 집안 정리, 코딩 테스트 문제 푼 거 블로그에 올리기, 바닥 청소, 책 읽기, 집안 정리.....

 

(url: https://programmers.co.kr/learn/courses/30/lessons/12950)

 

728x90