아무나개발하자

행렬의 덧셈 본문

코딩테스트

행렬의 덧셈

개발천재나천재 2023. 1. 7. 09:03

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12950

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

풀이

'''
리스트 출력 방법
이차원 배열 a가 있다고 가정할때,
[[0, 0], 
[0, 0]]

(1)
for i in a:        # a에서 안쪽 리스트를 꺼냄
    for j in i:    # 안쪽 리스트에서 요소를 하나씩 꺼냄
        print(j, end=' ')
        
(2)
for i in range(len(a)):            \
    for j in range(len(a[i])):  
            print(a[i][j], end=' ')
            
(3)
이차원 배열 선언방법
answer = [
        [0] * len(a[i])
        for i in range(len(a))
]

좋은 풀이 방법 

def solution(arr1, arr2):
    answer = []

    for a, b in zip(arr1, arr2):
        l = []
        for x, y in zip(a, b):
            l.append(x + y)
        answer.append(l)

    return answer
'''
def solution(arr1, arr2):
    answer = [
        [0] * len(arr1[i])
        for i in range(len(arr1))
    ]
    
    for i in range(len(arr1)):
        for j in range(len(arr1[i])):
            answer[i][j] += arr1[i][j] + arr2[i][j]
            
    return answer

 

 

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12950

'코딩테스트' 카테고리의 다른 글

시저 암호  (0) 2023.01.09
다음 큰 숫자  (0) 2023.01.08
프린터  (0) 2023.01.05
H-Index  (0) 2023.01.04
멀리 뛰기  (0) 2023.01.04