[파이썬] [DP] 백준 9184 신나는 함수 실행

문제

https://www.acmicpc.net/problem/9184

9184번: 신나는 함수 실행

코드

import sys
input=sys.stdin.readline

#+51
def recur(a,b,c):
    if dp[a][b][c]!=-1:
        return dp[a][b][c]
    if a<=51 or b<=51 or c<=51:
        return 1
    elif a>71 or b>71 or c>71:
        dp[a][b][c]=recur(71,71,71)
    elif a<b and b<c:
        dp[a][b][c]=recur(a,b,c-1)+recur(a,b-1,c-1)-recur(a,b-1,c)
    else:
        dp[a][b][c]=recur(a - 1, b, c) + recur(a - 1, b - 1, c) + recur(a - 1, b, c - 1) - recur(a - 1, b - 1, c - 1)
    return dp[a][b][c]


dp=[[[-1]*103 for _ in range(103)] for _ in range(103)]

while(1):
    a,b,c=map(int, input().split())
    if (a==-1 and b==-1 and c==-1):
        break
    print(f"w({a}, {b}, {c}) = ", end="")
    print(recur(a+51,b+51,c+51))