Matrix Sum
Programming Interview
Medium
8 views
Problem Description
Given {x}, Create Matrix class with add method. Output resulting matrix.
Input Format
First line r c. Next r lines matrix A. Next r lines matrix B.
Output Format
Matrix sum.
Sample Test Case
Input:
2 2
1 2
3 4
10 20
30 40
Official Solution
import sys
lines=sys.stdin.read().splitlines()
if not lines: sys.exit(0)
r,c=map(int,lines[0].split())
A=[]
B=[]
idx=1
for _ in range(r):
A.append(list(map(int,(lines[idx] if idx<len(lines) else '').split()[:c]))); idx+=1
for _ in range(r):
B.append(list(map(int,(lines[idx] if idx<len(lines) else '').split()[:c]))); idx+=1
class Matrix:
def __init__(self,mat):
self.mat=mat
def add(self,other):
res=[]
for i in range(len(self.mat)):
row=[]
for j in range(len(self.mat[0])):
row.append(self.mat[i][j]+other.mat[i][j])
res.append(row)
return Matrix(res)
M=Matrix(A).add(Matrix(B)).mat
out=[]
for row in M:
out.append(' '.join(map(str,row)))
print('\
'.join(out))
Solutions (0)
No solutions submitted yet. Be the first!
No comments yet. Start the discussion!