搜索--递归--记忆化搜索

题目描述

对于一个递归函数 w(a,b,c)

• 如果 a≤0 或 b≤0 或 c≤0 就返回值11。
• 如果 a>20 或 b>20 或 c>20 就返回 w(20,20,20)
• 如果 a<b 并且 b<c 就返回 w(a,b,c−1)+w(a,b−1,c−1)−w(a,b−1,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）
• 注意：w(30,−1,0) 又满足条件 11 又满足条件 22，请按照最上面的条件来算，答案为 11。

• 输入格式

  会有若干行。

  并以 −1,−1,−1−1,−1,−1 结束。

  输出格式

  输出若干行，每一行格式：

  w(a, b, c) = ans

  注意空格。

做题思路

这是个简单的递归函数，但实现起来可能会有些问题。

当 a,b,c 均为 15 时，要计算w(14,15,15)+w(14,14,15)+w(14,15,14)−w(14,14,14）

=w(13,15,15)+w(13,14,15)+w(13,15,14)−w(13,13,14）+w(13,14,15)+w(13,13,15)+w(13,14,14)−w(13,13,14）.......

可以看到，在算这个函数时，没有保留中间数据，再要重复算一个函数时，之前算过而现在还要在算一遍，这就导致了重复里很多相同步骤，增大了时间复杂度。因此我们必须存储中间数据，避免重复讲计算，那么自然而然就想到了用记忆化搜索。

具体实现

首先要建立一个数组存储中间数据

long long a,b,c,ans[22][22][22]; //设置3个参数和存储的数组
bool ansb[22][22][22];//判断是否计算过

然后就是核心的递归部分先判断是否满足第一个条件

if(a<=0||b<=0||c<=0) return 1;//是否满足第一个条件

然后是第二个条件

if(a>20 or b>20 or c>20) return w(20,20,20);

 特判结束后就是主要的计算部分

if(ansb[a][b][c]) return ans[a][b][c];//先判断是否算过，算过的话直接返回，避免重复计算
else
{
	if(a<b&&b<c)//条件三
	{
		ans[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
        //先用递归计算ans[a][b][c]（具体内容不用管，让他自己跑）
		ansb[a][b][c]=1;//标记此数已经算过
		return ans[a][b][c];//返回
	}
	else//条件四
	{
		ans[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);
		ansb[a][b][c]=1;
		return ans[a][b][c];//与上面同理
	}
}

然后解决输入输出，由于是无限输入，三数均为-1停止因此采用while（1）循环

 while(1)//无限循环
	{
        scanf("%lld%lld%lld",&a,&b,&c);//输入
		if(a==-1 and b==-1 and c==-1) break;//判断终止条件
		printf("w(%lld, %lld, %lld) = %lld\n",a,b,c,w(a,b,c));//输出
	}

那么这个题就算做完了，以下是完整代码

完整代码

#include<bits/stdc++.h>
using namespace std;
long long a,b,c,ans[22][22][22]; //设置3个参数和存储的数组
bool ansb[22][22][22];//判断是否计算过
long long w(long long a,long long b,long long c)
{
	if(a<=0||b<=0||c<=0) return 1;//是否满足第一个条件
	if(a>20 or b>20 or c>20) return w(20,20,20);
	if(ansb[a][b][c]) return ans[a][b][c];//先判断是否算过，算过的话直接返回，避免重复计算
	else
	{
		if(a<b&&b<c)//条件三
		{
			ans[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
	        //先用递归计算ans[a][b][c]（具体内容不用管，让他自己跑）
			ansb[a][b][c]=1;//标记此数已经算过
			return ans[a][b][c];//返回
		}
		else//条件四
		{
			ans[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);
			ansb[a][b][c]=1;
			return ans[a][b][c];//与上面同理
		}
	}
}
int main()
{
	 while(1)//无限循环
	{
        scanf("%lld%lld%lld",&a,&b,&c);//输入
		if(a==-1 and b==-1 and c==-1) break;//判断终止条件
		printf("w(%lld, %lld, %lld) = %lld\n",a,b,c,w(a,b,c));//输出
		//这里一定要注意中间有空格！！！
		//这里一定要注意中间有空格！！！
		//这里一定要注意中间有空格！！！
	}
}