[백준/BOJ/C++] 2738번 행렬 덧셈

[문제]

N*M크기의 두 행렬 A와 B가 주어졌을 때, 두 행렬을 더하는 프로그램을 작성하시오.

 

[입력]

첫째 줄에 행렬의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 차례대로 주어진다. 이어서 N개의 줄에 행렬 B의 원소 M개가 차례대로 주어진다. N과 M은 100보다 작거나 같고, 행렬의 원소는 절댓값이 100보다 작거나 같은 정수이다.

 

[출력]

첫째 줄부터 N개의 줄에 행렬 A와 B를 더한 행렬을 출력한다. 행렬의 각 원소는 공백으로 구분한다.

 

[Source Code]

#include <stdio.h>
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <sstream>
using namespace std;


int main() {
	int n, m;
	cin >> n >> m;

	int *arr = new int[n*m];

	for(int i = 0; i < n*m; i++){
		int input;
		cin >> input;
		arr[i] = input;
	}
	for(int i = 0; i < n*m; i++){
		int input;
		cin >> input;
		arr[i] += input;
	}
	
	for(int i = 0; i < n*m; i += m){
		for(int j = 0; j < m; j++){
			cout << arr[i+j] <<" "; 
		}
		cout << "\n";
	}
}

[결과 화면]