행렬 덧셈 C++

📝 문제

https://www.acmicpc.net/problem/2738

 

문제 설명

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

---

입력

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

---

출력

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

---

입출력

예제 입력 1

3 3
1 1 1
2 2 2
0 1 0
3 3 3
4 4 4
5 5 100

예제 출력 1

4 4 4
6 6 6
5 6 100

---

풀이 코드

#include<iostream>
using namespace std;

int main()
{
	int row(0), col(0);
	cin >> row >> col;

	int** aArray = new int* [row] {};
	int** bArray = new int* [row] {};

	for (int i = 0; i < row; ++i)
	{
		aArray[i] = new int[col] {};
		bArray[i] = new int[col] {};
	}

	for (int i = 0; i < row; ++i)
	{
		for (int j = 0; j < col; ++j)
			cin >>aArray[i][j];
	}

	for (int i = 0; i < row; ++i)
	{
		for (int j = 0; j < col; ++j)
			cin >> bArray[i][j];
	}

	for (int i = 0; i < row; ++i)
	{
		for (int j = 0; j < col; ++j)
			cout << aArray[i][j] + bArray[i][j] << ' ';

		cout << endl;
	}

	// 2차원 동적 배열 반환
	for (int i = 0; i < row; ++i)
	{
		delete[] aArray[i];
		delete[] bArray[i];
	}

	delete[] aArray;
	delete[] bArray;
	return 0;
}

• 이 문제를 해석해보면, 단순한 행렬의 덧셈이다.
  • A행렬과 B행렬의 행과 열은 입력받은 값을 똑같이 사용하기 때문에 행렬 덧셈이 가능하다.

✍️ 중요 포인트

• 2차원 동적 배열을 사용해 유동적으로 크기를 정해주었다.
• 2차원 동적 배열의 메모리 반환도 꼼꼼히 해주어야 한다.