[백준/BOJ/알고리즘/파이썬(python)]#9148:신나는 함수 실행[재귀]

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

9184번: 신나는 함수 실행

 

재귀 호출을 하는 문제로 

1. 리스트배열 할당을 20만큼씩 (20이 넘으면 어차피 조건문에서 걸리므로) 하는 것

2. 재귀배열이 있다면 재귀배열을 리턴, 아니면 구하는 조건문으로 넘어가는 것

 

이 두 가지를 신경써야한다. 

 

파이썬 코드 

import sys
sys.stdin.readline

arr = [[[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 arr[a][b][c]:
        return arr[a][b][c]
    if a<b and b<c:
        arr[a][b][c] = w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c)
        return arr[a][b][c]
    else:
        arr[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 arr[a][b][c]


while 1:
    a,b,c = map(int,input().split())
    res = w(a,b,c)
    if a==-1 and b==-1 and c==-1:
        break
    print(f'w({a}, {b}, {c}) = {res}')