给定一个函数 f(a, b, c):
如果 a ≤ 0 或 b ≤ 0 或 c ≤ 0 返回值为 1;
如果 a > 20 或 b > 20 或 c > 20 返回值为 f(20, 20, 20);
如果 a < b 并且 b < c 返回 f(a, b, c−1) + f(a, b−1, c−1) − f(a, b−1, c);
其它情况返回 f(a−1, b, c) + f(a−1, b−1, c) + f(a−1, b, c−1) − f(a-1, b-1, c-1)。
看起来简单的一个函数?你能做对吗?
Input
输入包含多组测试数据,对于每组测试数据:
输入只有一行为 3 个整数a, b, c(a, b, c < 30)。
Output
对于每组测试数据,输出函数的计算结果。
Example Input
1 1 1 2 2 2
Example Output
2 4
所谓记忆化搜索,其实是dp的递归式写法,只不过为了避免大量多余的计算而把计算结果记录下来,这道题在当时学动归的时候知道怎么敲,但是不太理解,现在看来,别有一番理解啊
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = 30;
int f[N][N][N]; //将计算结果记录在数组中,避免重复计算
int rec(int a, int b, int c)
{
if(a <= 0 || b <= 0 || c <= 0) return 1;
if(f[a][b][c] != -1) //数组里已经有计算结果,直接返回
return f[a][b][c];
if(a > 20 || b > 20 || c > 20)
return rec(20, 20, 20);
if(a < b && b < c)
return f[a][b][c] = rec(a, b, c-1)+rec(a, b-1, c-1)-rec(a, b-1, c); //将结果记录在数组中
return f[a][b][c] = rec(a-1, b, c)+rec(a-1, b-1, c)+rec(a-1, b, c-1)-rec(a-1, b-1, c-1);
}
int main()
{
int a, b, c;
memset(f, -1, sizeof(f)); //只需要初始化一次就够了,f数组会一直记录所得到的计算结果
while(~scanf("%d%d%d", &a, &b, &c))
{
printf("%d\n", rec(a, b, c));
}
return 0;
}