洛谷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;
}
参考大佬们的题解写的,根本没想到记忆化搜索,以前也没怎么接触过,希望以后能记住,加油加油