행렬 덧셈 풀이

문제 링크: https://www.acmicpc.net/problem/2738

백준 2차원 배열 1단계 - 2738번 행렬 덧셈를 풀어보았다.

 

풀이: 배열을 만들고 입력값을 받아 더한다.

 

C++

#include <iostream>

using namespace std;
int main()
{
  int n,m;
  cin>>n>>m;
  int arr[101][101]={0,};
  for(int k=0;k<2;k++)
  { 
  	for(int i=0;i<n;i++)
    {
    	for(int j=0;j<m;j++)
        {
			int inp;
			cin>>inp;
			arr[i][j]+=inp;
		}
	}
  } 
  for(int i=0;i<n;i++)
  {
  	for(int j=0;j<m;j++)
    {
    	cout<<arr[i][j]<<" ";
    }
    cout<<"\n";
  }
}

Python

n,m=map(int,input().split())
mat=[[0 for i in range(m)]for j in range(n)] # 0으로 초기화한 2차원 배열을 만든다
for k in range(2):
	for i in range(n):
		tem=list(map(int,input().split()))
		for j in range(m):
			mat[i][j]+=tem[j] # 배열에 입력값을 받아 더해준다
for i in range(n):
	for j in range(m):
		print(mat[i][j],end=' ')
	print('')

Java