BOJ 9184 [신나는 함수 실행]

BOJ 신나는 함수 실행 : www.acmicpc.net/problem/9184

9184번: 신나는 함수 실행

신나지 않은 함수 실행이다.

 

메모이제이션을 사용해 풀었다.

 

배열 인덱스에 50을 더한 이유는 마이너스도 있으니깐,,

 

밑에 코드는 +50때문에 눈이 좀 피곤할 수도 있다.

 

#include <iostream>
#include <string>
 
using namespace std;
 
int arr[102][102][102];
bool visited[102][102][102];
 
int w(int a, int b, int c)
{
    if (visited[a + 50][b + 50][c + 50])
    {
        return arr[a + 50][b + 50][c + 50];
    }
 
    if (a <= 0 || b <= 0 || c <= 0)
    {
        return 1;
    }
 
    if (a > 20 || b > 20 || c > 20)
    {
        int res = w(20, 20, 20);
        visited[70][70][70] = true;
        arr[70][70][70] = res;
        return res;
    }
 
    if (a < b && b < c)
    {
        int x = w(a, b, c - 1);
        visited[a + 50][b + 50][c - 1 + 50] = true;
        arr[a + 50][b + 50][c - 1 + 50] = x;
 
        int y = w(a, b - 1, c - 1);
        visited[a + 50][b - 1 + 50][c - 1 + 50] = true;
        arr[a + 50][b - 1 + 50][c - 1 + 50] = y;
        
        int z = w(a, b - 1, c);
        visited[a + 50][b - 1 + 50][c + 50] = true;
        arr[a + 50][b - 1 + 50][c + 50] = z;
 
        return x + y - z;
    }
 
    int x = w(a - 1, b, c);
    visited[a - 1 + 50][b + 50][c + 50] = true;
    arr[a - 1 + 50][b + 50][c + 50] = x;
 
    int y = w(a - 1, b - 1, c);
    visited[a - 1 + 50][b - 1 + 50][c + 50] = true;
    arr[a - 1 + 50][b - 1 + 50][c + 50] = y;
 
    int z = w(a - 1, b, c - 1);
    visited[a - 1 + 50][b + 50][c - 1 + 50] = true;
    arr[a - 1 + 50][b + 50][c - 1 + 50] = z;
 
    int k = w(a - 1, b - 1, c - 1);
    visited[a - 1 + 50][b - 1 + 50][c - 1 + 50] = true;
    arr[a - 1 + 50][b - 1 + 50][c - 1 + 50] = k;
 
    return x + y + z - k;
}
 
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(0);
 
    while (true)
    {
        int a, b, c;
 
        cin >> a >> b >> c;
 
        if (a == -1 && b == -1 && c == -1)
        {
            break;
        }
        int result = w(a, b, c);
        cout << "w(" << a << ", " << b << ", " << c << ") = " << w(a, b, c) << endl;
    }
 
    return 0;
}