洛谷Function

在刷洛谷的时候遇到了一到需要用到记忆化搜索的题

看起来是很简单的递归,但是题目中说了a,b,c均为15时调用的次数将非常的多,也就是说这个递归函数会重复计算a,b,c函数的值,为了避免这种情况,我们使用记忆化数组保存每个a,b,c所对应的函数值,当出现对应的a,b,c时,如果数组中有,就可以直接从数组中拿出数值,如果没有,则把计算出来的数据存入数组

贴一下代码,这个我觉得逻辑是没有问题的,但是可能因为一些细节问题导致全部wa了…
有时间再来搞一下

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
ll dp[25][25][25];

ll w(ll a,ll b,ll c);

int main(){
    while(true){
        ll a,b,c;
        scanf("%lld%lld%lld",&a,&b,&c);
        if(a == -1 && b == -1 && c == -1){
            break;
        }
        memset(dp,0,sizeof(dp));
        printf("w(%lld, %lld, %lld) = ",a,b,c);
        if(a > 20 ) a = 21;
        if(b > 20 ) b = 21;
        if(c > 20 ) c = 21;
        printf("%lld",w(a,b,c));
    }
    return 0;    
}

ll w(ll a,ll b,ll c){
    if(a<= 0 || b <=0 || c <= 0){
        return 1;
    }
    else if(dp[a][b][c] != 0){
        return dp[a][b][c];
    }
    else if(a > 20 || b > 20 || c > 20){
        dp[a][b][c] = w(20,20,20);
    }

    else if(a<b && b < c){
       dp[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
    }
    else{
        dp[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 dp[a][b][c];
}

这种用法我们要注意一下返回值,如果数组中有,我们就直接拿出来返回,如果没有,我们就要先把它存进去,再返回(记住一定要返回阿不然递归就不存在了…)