컴퓨터/알고리즘 풀이 (PS) (C++)

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

플레이스의 이중생활 2022. 8. 1. 01:11
반응형

걸린 시간: 25분

메모라이제이션을 사용해야하는 문제였다.
처음에는 뭔 코드인지 햇갈렸으나, 대충 해석해서 짜본후, 결과값 맞는 걸 확인한후, 시간 오래걸리는 문제 해결하기위해, 각 값들을 3차원 배열에 저장해 바로바로 꺼내올수 있게 만들었다.
중간에 함수의 값을 저장한 배열을 어느 타이밍에 불러올지가 애매했지만, 문제에 맞지 않는 값들을 먼저 거른 후 처리한다고 생각하니 위치가 정확히 정해졌다.

dp는 거의 안해봐서 솔직히 지금은 재귀랑 큰 차이점을 못느끼겠다.

#include <iostream>
#include <vector>

using namespace std;
int s[22][22][22] = {0,};
int w(int a, int b, int c)
{
    int at, bt, ct,dt;
    if (a <= 0 || b <= 0 || c <= 0)
    {
        return 1;
    }
    if (a > 20 || b > 20 || c > 20)
    {
        return w(20, 20, 20);
    }
    if (s[a][b][c] == 0)
    {
        if (a < b && b < c)
        {
            at = w(a, b, c - 1);
            bt = w(a, b - 1, c - 1);
            ct = w(a, b - 1, c);
            s[a][b][c - 1] = at;
            s[a][b-1][c - 1] = bt;
            s[a][b-1][c] = ct;
            return  at + bt - ct;
        }
        else
        {
            at = w(a - 1, b, c);
            bt = w(a - 1, b - 1, c);
            ct = w(a - 1, b, c - 1);
            dt = w(a - 1, b - 1, c - 1);
            s[a-1][b][c] = at;
            s[a-1][b - 1][c] = bt;
            s[a-1][b][c-1] = ct;
            s[a-1][b - 1][c-1] = dt;

            return at + bt + ct - dt;
        }
    }
    else
    {
        return s[a][b][c];
    }
}
int main()
{
    while (1)
    {
        int a, b, c;
        cin >> a >> b >> c;
        if (a == -1 && b == -1 && c == -1)
        {
            break;
        }
        printf("w(%d, %d, %d) = %d\n", a, b, c, w(a, b, c));

    }
}


반응형