[BOJ] #9184. 신나는 함수 실행 / 실버2 / 60분 / 성공

Author: Mingguriguri

minjeong/DFSBFS/2024-07-26-[백준]-#13305-주유소.py

@@ -0,0 +1,23 @@
+import sys
+input = sys.stdin.readline
+
+def w(a, b, c):
+    if (a, b, c) in memo:
+        return memo[(a, b, c)]
+    if a <= 0 or b <= 0 or c <= 0:
+        memo[(a, b, c)] = 1
+        return 1
+    elif a > 20 or b > 20 or c > 20:
+        memo[(a, b, c)] = w(20, 20, 20)
+    elif a < b and b < c:
+        memo[(a, b, c)] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c)
+    else:
+        memo[(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 memo[(a, b, c)]
+
+memo = {}
+while True:
+    a, b, c = map(int, input().split())
+    if a == -1 and b == -1 and c == -1:
+        break
+    print('w(%d, %d, %d) = %d' % (a, b, c, w(a, b, c)))