记忆化宏定义

记忆化宏定义

在我们编写递归函数的时候，经常要用到记忆化搜索。我们可以使用一个宏定义来替换记忆化处理。这样记忆化处理就不会侵染我们的逻辑代码。类似于代理函数的思想。

P1464

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

#define FR freopen("in.txt", "r", stdin)
#define FW freopen("out.txt", "w", stdout)

#define FXMEM(x, y, z) (mem[x][y][z] ? mem[x][y][z] : mem[x][y][z] = w(x, y, z))

ll mem[25][25][25];

ll w(ll a, ll b, ll c)
{
    if (a <= 0 || b <= 0 || c <= 0)
    {
        return 1;
    }
    else if (a > 20 || b > 20 || c > 20)
    {
        return FXMEM(20, 20, 20);
    }
    else if (a < b && b < c)
    {
        return FXMEM(a, b, c - 1) + FXMEM(a, b - 1, c - 1) - FXMEM(a, b - 1, c);
    }
    else
    {
        return FXMEM(a - 1, b, c) + FXMEM(a - 1, b - 1, c) + FXMEM(a - 1, b, c - 1) - w(a - 1, b - 1, c - 1);
    }
}

int main()
{
    ll a, b, c;
    while (scanf("%lld %lld %lld", &a, &b, &c) != EOF && !(a == -1 && b == -1 && c == -1))
    {
        printf("w(%lld, %lld, %lld) = %lld\n", a, b, c, w(a, b, c));
    }
    return 0;
}

值得注意的是，宏只需要在递归里替换即可，外部调用不用替换，主要是避免数组下标越界。