[백준 9184번] 신나는 함수 실행 C++

www.acmicpc.net/problem/9184

9184번: 신나는 함수 실행

그냥 간단한 메모이제이션 문제

 

 

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
 
int dp[51][51][51];
 
int w(int a, int b, int c) {
 
 
    if (a <= 0 || b <= 0 || c <= 0)
        return 1;
    if (dp[a][b][c] != 0) return dp[a][b][c];
    if (a > 20 || b > 20 || c > 20) {
        if (dp[20][20][20] != 0)
            return dp[20][20][20];
        return w(20, 20, 20);
    }
    else if (a < b && b < c)
        return dp[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
    else
        return 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);
}
int main() {
 
    while (1) {
        int z, x, v;
        cin >> z >> x >> v;
        if (z == -1 && x == -1 && v == -1)
            break;
        cout << "w(" << z << ", " << x << ", " << v << ") = " << w(z, x, v) << "\n";
 
    }
    return 0;
}