[Python] 백준 #9184- 신나는 함수 실행

문제

---

9184번: 신나는 함수 실행

코드

---

My answer

import sys
input=sys.stdin.readline

dp=[]
for i in range(101):
    dp.append([])
    for j in range(101):
        dp[i].append([])
        for k in range(101):
            dp[i][j].append([0])

dp[20][20][20]=1048576

for i in range(-50,51,1):
    for j in range(-50,51,1):
        for k in range(-50,51,1):
            if (i <= 0 or j <= 0 or k <= 0):
                dp[i][j][k]=1
            elif (i > 20 or j > 20 or k > 20):
                dp[i][j][k]=dp[20][20][20]
            elif (i < j and j < k):
                dp[i][j][k]=dp[i][j][k-1]+dp[i][j-1][k-1]-dp[i][j-1][k]
            else:
                dp[i][j][k]=dp[i-1][j][k]+dp[i-1][j-1][k]+dp[i-1][j][k-1]-dp[i-1][j-1][k-1]
while(1):
    tmp=list(map(int,input().split()))
    if(tmp==[-1,-1,-1]):
        break
    else:
        print("w({0}, {1}, {2}) = {3}".format(tmp[0],tmp[1],tmp[2],dp[tmp[0]][tmp[1]][tmp[2]]))

Another answer

import sys
input = sys.stdin.readline

dp = {}

def w(a, b, c):
	if (a, b, c) in dp.keys():
		return dp[(a, b, c)]
	else:
		if a <= 0 or b <= 0 or c <= 0: return 1
		elif a > 20 or b > 20 or c > 20: return w(20, 20, 20)
		else:
			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 == b == c == -1: break
	print(f'w({a}, {b}, {c}) = {w(a, b, c)}')

풀이

우선 나는 어떻게 풀지 감이 안와서 그냥 문제그대로를 3중반복문 돌려서 풀었다. 이렇게 풀라고 의도한건 아니겠지만 그냥 풀렸다. 다만 저렇게 할 때 맨처음 틀렸었는데 그 이유는 나머지는 i,j,k,가 이미 구해진것들로 계산하기 때문에 상관없는데 저기 두번째 조건에서 dp[20][20][20]가 아직 안구해졌는데 대입을 해버리면서 오류가 나서 임의로 맨위에 dp[20][20][20] 값을 대입해줬다. 그냥 끼워맞춘 코드 같다. 

아래 코드는 엄청 신기했다... 재귀를 쓰는데 dp를 딕셔너리로 만들어서 key를 튜플형식으로 넣어서 처리한 것같다..대단하다...