9184_신나는함수실행

🗨️Comment

• return문에서 함수를 반복적으로 호출하지말고, visited배열에 담아두고, 그 값을 return
• if분기문 중, 음수인 경우 다음에 visited배열이 존재하는 경우 바로 return해주기

⚠️ Error

• 시간초과 if분기문에서 호출하는 모든 w함수들에 대해서 visited배열로 받은 다음, 계산된 visited값을 리턴해서 해결
  
• 틀렸습니다

  if a <= 0 or b <= 0 or c <= 0:
          # visited[a][b][c] = 1 # <- 음수인 경우도 포함 & 이땐 바로 1만 리턴하면 됨
          return 1

---

🥳 정답코드

'''
23.05.02
11:15 ~ 11 : 55 
'''
#  1초 / 128MB
# Error : 시간초과 / 틀림 

import sys
input = sys.stdin.readline

visited = [[[0] * (51) for _ in range(51)] for _ in range(51)]

def w(a, b, c):
    # 셋중 하나라도 음수면, 무조건 1을 리턴
    # 셋중 하나라도 20초과면 무조건 W(20, 20, 20)    

    if a <= 0 or b <= 0 or c <= 0:
        return 1
    
    elif visited[a][b][c] != 0:
        return visited[a][b][c]
    
    elif a > 20 or b > 20 or c > 20:
        visited[a][b][c] = w(20, 20, 20)
        return visited[a][b][c]

    elif a < b and b < c:
        visited[a][b][c] = w(a, b, c-1) + w(a, b-1, c-1) - w(a, b-1, c)
        return visited[a][b][c]

    else:
        visited[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 visited[a][b][c]

# a, b, c: 각각 -50이상, 50이하 
while True:
    a, b, c = map(int, input().split())

    # terminal condition 
    if a == -1 and b == -1 and c == -1:
        break 

    # Solution 
    res = w(a, b, c) 

    # Output
    print("w({0}, {1}, {2}) = {3}".format(a, b, c, res))