출처 : programmers.co.kr/learn/courses/30/lessons/12950
* 이 포스팅은 Programmers 코딩 테스트 연습 Level 1 "행렬의 덧셈" 문제풀이입니다.
문제 원본은 링크를 클릭하시면 확인하실 수 있습니다.
# 문제 설명
-
행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다.
-
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]] |
# 코드 설명
1
2
3
4
5
6
7
8
|
def solution(arr1, arr2):
answer = []
for i in range(len(arr1)):
tmp = []
for j in range(len(arr1[i])):
tmp.append(arr1[i][j] + arr2[i][j])
answer.append(tmp)
return answer
|
cs |
'Python > Programmers' 카테고리의 다른 글
[Python/Programmers/Level 2] 행렬의 곱셈 (0) | 2020.10.06 |
---|---|
[Python/Programmers/Level 2] 피보나치 수 (0) | 2020.10.06 |
[Python/Programmers/Level 2] 다리를 지나는 트럭 (0) | 2020.10.05 |
[Python/Programmers/Level 2] 기능개발 (0) | 2020.10.04 |
[Python/Programmers/Level 1] 문자열 내 p와 y의 개수 (0) | 2020.10.01 |