백준 알고리즘 9184번 신나는 함수 실행(python)

기본적인 다이나믹 프로그래밍 & 재귀 문제이다.

dp = [[[0]*21 for a in range(21)] for b 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 dp[a][b][c]:
        return dp[a][b][c]
    if a < b < c:
        dp[a][b][c] = w(a, b, c-1) + w(a, b-1, c-1) - w(a, b-1, c)
        return dp[a][b][c]
    dp[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 dp[a][b][c]

while 1:
    a, b, c = map(int, input().split())
    if a == b == c == -1:
        break
    print("w(%d, %d, %d) = %d"%(a, b, c, w(a, b, c)))