map有关的运算符重载

Function

题目描述

对于一个递归函数 $w(a,b,c)$

• 如果 $a\le 0$ 或 $b\le 0$ 或 $c\le 0$ 就返回值$ 1$。
• 如果 $a>20$ 或 $b>20$ 或 $c>20$ 就返回 $w(20,20,20)$
• 如果 $a<b$ 并且 $b<c$ 就返回$ w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c)$。
• 其它的情况就返回 $w(a-1,b,c)+w(a-1,b-1,c)+w(a-1,b,c-1)-w(a-1,b-1,c-1)$

这是个简单的递归函数，但实现起来可能会有些问题。当 $a,b,c$ 均为 $15$ 时，调用的次数将非常的多。你要想个办法才行。

注意：例如 $w(30,-1,0)$ 又满足条件 $1$ 又满足条件 $2$，请按照最上面的条件来算，答案为 $1$。

输入格式

会有若干行。

并以 $-1,-1,-1$ 结束。

保证输入的数在 $[-9223372036854775808,9223372036854775807]$ 之间，并且是整数。

输出格式

输出若干行，每一行格式：

w(a, b, c) = ans

注意空格。

样例 #1

样例输入 #1

1 1 1
2 2 2
-1 -1 -1

样例输出 #1

w(1, 1, 1) = 2
w(2, 2, 2) = 4

题解：

这个题，使用记忆化搜索就好了
然后用自定义数据类型的时候需要重载小于号，重载方式在代码里了

#include <iostream>
#include <map>
#define ll long long
using namespace std;

struct th {
    ll a, b, c;
    th(ll aa, ll bb, ll cc) :a(aa), b(bb), c(cc) {}
    bool operator<(const th& t)const {
        return a < t.a || (a == t.a && b < t.b) || (a == t.a && b == t.b && c < t.c);
    }
};
map<th, ll> vis;

ll w(ll a, ll b, ll c) {
    if (a <= 0 || b <= 0 || c <= 0) {
        return 1;
    }
    else if (vis.count(th(a, b, c))) {
        return vis[th(a, b, c)];
    }
    else if (a > 20 || b > 20 || c > 20) {
        return w(20, 20, 20);
    }
    else if (a < b && b < c) {
        vis[th(a, b, c)] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
    }
    else
        vis[th(a, b, c)] = 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 vis[th(a, b, c)];
}

int main()
{
    int t1, t2, t3;
    cin >> t1 >> t2 >> t3;
    while (!(t1 == -1 && t2 == -1 && t3 == -1)) {
        cout << "w(" << t1 << ", " << t2 << ", " << t3 << ") = " << w(t1, t2, t3) << '\n';
        cin >> t1 >> t2 >> t3;
    }
    return 0;
}