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

www.acmicpc.net/problem/9184

9184번: 신나는 함수 실행

DP만 이용한다면 쉽게 풀 수 있는 문제입니다. 문제의 점화식이 다 주어지므로 그대로 코딩하시면 됩니다.

#include<iostream>

using namespace std;

#define MAX 21

int DP[MAX][MAX][MAX];

int w(int a, int b, int c)
{
    if (a <= 0 || b <= 0 || c <= 0) return 1;
    if (a > 20 || b > 20 || c > 20) return w(20, 20, 20);
    if (DP[a][b][c]) return DP[a][b][c];
    if (a < b && b < c) DP[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
    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];
}

int main(void)
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int a, b, c; cin >> a >> b >> c;

    while (a != -1 || b != -1 || c != -1) {
        cout << "w(" << a << ", " << b << ", " << c << ") = " << w(a, b, c) << "\n";
        cin >> a >> b >> c;
    }

    return 0;
}

다만 a, b, c 값 중 하나가 20이 넘을 경우 아래 코드처럼 함수를 호출하기 전에 변숫값을 바꿔줘 봤는데 출력은 정상적으로 되나 제출한 경우 틀렸습니다가 떴는데 어떤 부분 때문인지 잘 모르겠네요!

#include<iostream>

using namespace std;

#define MAX 21

int DP[MAX][MAX][MAX];

int w(int a, int b, int c)
{
    if (a <= 0 || b <= 0 || c <= 0) return 1;
    if (DP[a][b][c]) return DP[a][b][c];
    if (a < b && b < c) DP[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
    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];
}

int main(void)
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int a, b, c; cin >> a >> b >> c;

    while (a != -1 || b != -1 || c != -1) {
        cout << "w(" << a << ", " << b << ", " << c << ") = ";
        if (a > 20 || b > 20 || c > 20) {
            a = 20;
            b = 20;
            c = 20;
        }
        cout << w(a, b, c) << "\n";
        cin >> a >> b >> c;
    }

    return 0;
}