[알고리즘] 백준9184 신나는 함수 실행 - 파이썬

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

9184번: 신나는 함수 실행

 

 

 

[풀이 코드] 

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

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 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:
    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)}')

 

- 또 어렵게 생각하고 풀려다가 시간만 잡아먹은 문제이다.. 문제에 함수에 대한 코드가 다 있기 때문에 그대로 활용하면 됐었다.. 자꾸 오류가 나서 다른 분의 코드를 보며 공부했다.

 

 

 

[참고 블로그]

- https://upcount.tistory.com/8

백준 9184번 신나는 함수 실행 (Python)