递归的函数 记忆化搜索

给定一个函数 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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值