Python 9184 : 신나는 함수 실행 DP

문제

풀이

• 문제에서 주어진 재귀함수 식에 DP 배열을 사용해서
• 한 번 계산한 값을 DP 배열 저장했다가 필요하면 반환해주면 끝

잡담

arr = list([[0 for _ in range(21)] for _ in range(21)] for _ in range(21))

• 0 * 21 해도 되는데 for문 썼다
• 이게 실버2 수준은 아닌거 같다

전체 코드

import sys

r = sys.stdin.readline
arr = list([[0 for _ in range(21)] for _ in range(21)] for _ in range(21))


def w(a, b, c):
    if a <= 0 or b <= 0 or c <= 0:
        return 1
    if a > 20 or b > 20 or c > 20:
        return w(20, 20, 20)

    if arr[a][b][c]:
        return arr[a][b][c]

    if a < b < c:
        arr[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c)
        return arr[a][b][c]
    arr[a][b][c] = w(a - 1, b, c) + w(a - 1, b - 1, c) + w(a - 1, b, c - 1) - w(a - 1, b - 1, c - 1)
    return arr[a][b][c]


while True:
    a, b, c = map(int, r().split())
    if a == b == c == -1:
        break
    print(f'w({a}, {b}, {c}) = {w(a, b, c)}')