[Python] 부족한 금액 계산하기 & 행렬의 덧셈

 

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

프로그래머스

# 부족한 금액 계산하기
def solution(price, money, count):
    cnt_sum = count*(count+1) / 2
    price *= cnt_sum
    if money - price >0:
        return 0
    else:
        return (price-money)

# 다른 풀이
def solution(price, money, count):
    return max(0, count*(count+1)/2 * price - money)

 

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

프로그래머스

# 행렬의 덧셈
def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        i_row = []
        for j in range(len(arr1[0])):
            i_row.append(arr1[i][j]+arr2[i][j])
        answer.append(i_row)
    return answer

# 다른 풀이
def sumMatrix(A,B):
    return [list(map(sum, zip(*x))) for x in zip(A, B)]