map有关的运算符重载
Function
题目描述
对于一个递归函数 w ( a , b , c ) w(a,b,c) w(a,b,c)
- 如果 a ≤ 0 a \le 0 a≤0 或 b ≤ 0 b \le 0 b≤0 或 c ≤ 0 c \le 0 c≤0 就返回值$ 1$。
- 如果 a > 20 a>20 a>20 或 b > 20 b>20 b>20 或 c > 20 c>20 c>20 就返回 w ( 20 , 20 , 20 ) w(20,20,20) w(20,20,20)
- 如果 a < b a<b a<b 并且 b < c b<c 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 ) w(a-1,b,c)+w(a-1,b-1,c)+w(a-1,b,c-1)-w(a-1,b-1,c-1) 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 a,b,c a,b,c 均为 15 15 15 时,调用的次数将非常的多。你要想个办法才行。
注意:例如 w ( 30 , − 1 , 0 ) w(30,-1,0) w(30,−1,0) 又满足条件 1 1 1 又满足条件 2 2 2,请按照最上面的条件来算,答案为 1 1 1。
输入格式
会有若干行。
并以 − 1 , − 1 , − 1 -1,-1,-1 −1,−1,−1 结束。
保证输入的数在 [ − 9223372036854775808 , 9223372036854775807 ] [-9223372036854775808,9223372036854775807] [−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;
}