목차

신나는 함수 실행

ps
링크acmicpc.net/…
출처BOJ
문제 번호9184
문제명신나는 함수 실행
레벨실버 2
분류

DP

시간복잡도O(abc+t)
인풋사이즈a<=20, b<=20, c<=20, t<=?
사용한 언어Python
제출기록34000KB / 144ms
최고기록80ms
해결날짜2021/09/06

풀이

코드

"""Solution code for "BOJ 9184. 신나는 함수 실행".

- Problem link: https://www.acmicpc.net/problem/9184
- Solution link: http://www.teferi.net/ps/problems/boj/9184
"""

import functools
import sys


@functools.cache
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 a < b < c:
        return w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c)
    return (w(a - 1, b, c) + w(a - 1, b - 1, c) + w(a - 1, b, c - 1) -
            w(a - 1, b - 1, c - 1))


def main():
    while True:
        a, b, c = [int(x) for x in sys.stdin.readline().split()]
        if a == b == c == -1:
            break
        print(f'w({a}, {b}, {c}) = {w(a, b, c)}')


if __name__ == '__main__':
    main()