洛谷P1464 Function

题目
记忆化搜索通过特征值标记 a, b, c <= 20的结果。

#include <iostream>
#include <cmath>
#include <queue>
#include <map>
#include <cstring>
#include <algorithm>
#define flag a * 441 + b * 21 + c //特征值
using namespace std;

map<long long, long long> Map;

long long w(long long a, long long b, long long c)
{
    if (Map[flag]) {
        return Map[flag];
    }
    if (a <= 0 || b <= 0 || c <= 0) {
        return 1;
    }
    else if (a > 20 || b > 20 || c > 20) {
        return w(20, 20, 20);
    }
    else if (a < b && b < c) {
        Map[flag] =  w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
        return Map[flag];
    }
    else {
        Map[flag] = 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 Map[flag];
    }

}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    long long A, B, C;
    while (cin >> A >> B >> C && !(A == -1 && B == -1 && C == -1)) {
        long long ans = w(A, B, C);
        cout << "w(" << A << ", " << B << ", " << C << ") = " << ans << endl;
    }
    return 0;
}