[백준][동적 계획법1] 9184. 신나는 함수 실행 (파이썬/Python)

MAX = 21
dp=[[[0]*MAX for _ in range(MAX)] for _ in range(MAX)]

def w(a,b,c):
    if a<=0 or b<=0 or c<=0 <= 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: #a<b and 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 True: #-1, -1, -1이 나올 때까지 무한 loop로 실행

    a,b,c = map(int,input().split())

    if a== -1 and b == -1 and c == -1:
        break

    print(f'w{a,b,c} = {w(a,b,c)}')

< 오답 = 시간초과 뜸 >

#9184



dp = [[[0]*(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 dp[a][b][c]:
        return dp[a][b][c]
    if a<b<c:
      return w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,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 True:
    a,b,c = map(int,input().split())

    if a==-1 and b ==-1 and c==-1:
        break

    print(f'w{a,b,c} = {w(a,b,c)}')