[9184] 신나는 함수 실행

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

9184번: 신나는 함수 실행

<문제>

메모이제이션을 사용해보라는 문제로 생각된다. 의사코드를 문제에서 제공했기에, 그대로 옮겨주면 된다.

 

<소스코드>

#include <bits/stdc++.h>
using namespace std;
int memo[51][51][51];
int f(int a, int b, int c) {
    if (a <= 0 || b <= 0 || c <= 0) return 1;
    if (a > 20 || b > 20 || c > 20) return f(20, 20, 20);
    if (memo[a][b][c] != 0) return memo[a][b][c];
    if (a < b && b < c)
        return memo[a][b][c] =
                   f(a, b, c - 1) + f(a, b - 1, c - 1) - f(a, b - 1, c);
    return memo[a][b][c] = f(a - 1, b, c) + f(a - 1, b - 1, c) +
                           f(a - 1, b, c - 1) - f(a - 1, b - 1, c - 1);
}
int main(void) {
    while (1) {
        int x, y, z;
        cin >> x >> y >> z;
        if (x == -1 && y == -1 && z == -1) break;
        printf("w(%d, %d, %d) = %d\n", x, y, z, f(x, y, z));
    }
    return 0;
}