컴퓨터/알고리즘 풀이 (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));
}
}
반응형