Day7:P1464 Function C++记忆化搜索

 

代码思路： 

1、如果按照题目的意思直接递归，肯定会超时。所以我们可以采用记忆化搜索的思想

2、记忆化搜索，本质还是 动态规划，只是实现方式采用了 深度优先搜索 的形式，但是它不像 深度优先搜索那样 重复 枚举所有情况，而是把已经计算的子问题保存下来，这样就和动态规划的思想不谋而合了。

3、一些小的注意事项：（如果自己写的时候报错了，可以来看看）

（1）在本题中， a>20 或 b>20 或 c>20 就返回 w(20,20,20)；因此我们只需要记录0~20，数组大小可以为w[25][25][25]。

 (2) a≤0 或b≤0 或 c≤0 就返回值1（return 1);不可以写成return w[a][b][c],因为索引不可以为负数。

 (3)千万要避免数组越界问题。记住你开的数组大小，w[25][25][25]索引最大为24。所以当 a>20 或 b>20 或 c>20，需要先约束范围，避免越界。

代码实现如下：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
//typedef long long ll;
ll w[25][25][25] = {0}; //只需记忆0~20；因为如果a>20||b>20||c>20就返回w(20,20,20)
ll f(ll a, ll b, ll c) {//函数名不能用function（这是一个关键字）
	if (a <= 0 || b <= 0 || c <= 0) { //不能写成return w[a][b][c];因为索引不能为负数
		return 1;
	} else if (a > 20 || b > 20 || c > 20) {
		return f(20, 20, 20);
	} else if (w[a][b][c] != 0) { //这种情况必须放在a > 20 || b > 20 || c > 20之后，否则会造成w[25][25][25]越界
		return w[a][b][c];//如果已经被记忆过，直接返回值，不需要递归
	} else if (a < b && b < c) {
		return w[a][b][c] = f(a, b, c - 1) + f(a, b - 1, c - 1) - f(a, b - 1, c);
	} else {
		return w[a][b][c] = f(a - 1, b, c) + f(a - 1, b - 1, c) + f(a - 1, b, c - 1) - f(a - 1,
		                    b - 1, c - 1);
	}
}

int main() {
	ll a, b, c;
	while (scanf("%lld %lld %lld", &a, &b, &c)) {
		if (a == -1 && b == -1 && c == -1) {
			break;
		} else {
			printf("w(%lld, %lld, %lld) = %lld", a, b, c, f(a, b, c));
			printf("\n");
		}
	}
	return 0;
}