poj - 1579 - Function Run Fun（记忆化搜索）

题意：跟着题目中公式走。

题目链接：http://poj.org/problem?id=1579

——>>简单题。

#include <cstdio>
#include <cstring>

const int MAXN = 20 + 5;

int dp[MAXN][MAXN][MAXN];

void Init()
{
    memset(dp, -1, sizeof(dp));
}

int w(int a, int b, int c)
{
    if (a <= 0 || b <= 0 || c <= 0)
    {
        return 1;
    }
    else if (a > 20 || b > 20 || c > 20)
    {
        return w(20, 20, 20);
    }

    int& ans = dp[a][b][c];

    if (ans != -1)
    {
        return ans;
    }

    if (a < b && b < c)
    {
        ans = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
    }
    else
    {
        ans = 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 ans;
}

int main()
{
    int a, b, c;

    Init();
    while (scanf("%d%d%d", &a, &b, &c) == 3)
    {
        if (a == -1 && b == -1 && c == -1)
        {
            break;
        }
        printf("w(%d, %d, %d) = %d\n", a, b, c, w(a, b, c));
    }

    return 0;
}