每日一题2023/2/23 P1464 Function

题目描述：

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

1、a<=0 || b<=0 || c<=0 return 1;

2、a>20 || b>20 || c>20 return w(20,20,20);

3、a<b<c return w(a,b,c−1)+w(a,b−1,c−1)−w(a,b−1,c);

4、其他情况 return w(a−1,b,c)+w(a−1,b−1,c)+w(a−1,b,c−1)−w(a−1,b−1,c−1);

注意：满足多种情况时，按1>4的顺序来判断返回值

输入:a,b,c,要求输出w(a,b,c)的值

数据范围：

a,b,c均在int_max 范围内

题目分析：

观察给出的递归函数,虽然a,b,c的数字范围很大,但是只要任一超过20或者小于0就会有固定的返回值,所以我们只需要考虑在1-20之间的a,b,c组合,由于函数递归是让a,b,c从大到小递归,故w(a,b,c)>=1。

很明显我们需要用到记忆化的方法来存储一些已确定的w(a，b，c)的值，减少重复计算，因为返回值肯定大于0，所以用初始化为0即可判断是否计算过。由于a，b，c的数据范围缩小在1-20，所以只需要一q[25][25][25]大小的三维数组便足够了。

为了避免数组越界，我们则需要先进行条件1和2的判断，再来判断是否存入数组过即可

代码：

#include<iostream>
#include<cstdio>

using namespace std;

const int N = 25;
int q[N][N][N]={0};

int w(int a, int b, int c)
{
	if (a <= 0 || b <= 0 || c <= 0) return 1;
	else if (a > 20 || b > 20 || c > 20) return w(20, 20, 20);

	if (q[a][b][c]);
	else if (a < b && b < c) q[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
	else q[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 q[a][b][c];
}
int main()
{
	int a=0, b=0, c=0;
	while (true)
	{
		scanf("%d%d%d", &a, &b, &c);
		if (a == -1 && b == -1 && c == -1) break;
		printf("w(%d,%d,%d) = %d\n", a, b, c, w(a, b, c));
	}
	return 0;
}