洛谷P1464 Function

洛谷P1464 Function

一道非常巧妙的记忆化搜索
如果直接用题目中的描述写函数 直接超时 一个也过不了
在w函数中每次递归a，b，c都是在一个一个减的
有很多重复情况
如果每次计算得到了在该情况下w(a,b,c)的值都能存储起来
那么就可以节省很多不必要的计算 运行时间就会减少

#include<iostream>
using namespace std;
int dp[21][21][21]; //记录计算过的a，b，c的值
int vis[21][21][21]; //记录有没有算过
int w(long long a,long long b,long long c)
{
    if(a<=0||b<=0||c<=0) return 1;
    if(a>20||b>20||c>20) return w(20,20,20); //将a，b，c变小
    if(vis[a][b][c]==1) return dp[a][b][c];  //如果计算过了 就不用再算了
    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);
    vis[a][b][c] = 1;  //标记
    return dp[a][b][c];  //返回值
}
int main()
{
    long long a,b,c;
    cin>>a>>b>>c;
    while(a!=-1||b!=-1||c!=-1)
    {
        long long ans = w(a,b,c);
        printf("w(%lld, %lld, %lld) = %d\n",a,b,c,ans); //换行
        cin>>a>>b>>c;
    }

    return 0;
}

参考大佬们的题解写的，根本没想到记忆化搜索，以前也没怎么接触过，希望以后能记住，加油加油