P1464 Function(记忆化搜索)(宏定义)

发布时间 2023-12-18 18:09:16作者: 拍手称快

P1464 Function

记忆化搜索(引用洛谷上题解的原话,自己是个废物想不到):

只要确定好数组和范围就可以把搜索中的每一次结果存起来, 正在搜索的项目中也可以用到这些信息 。
这便是记忆化搜索的核心与关键:自定义函数与数组达到“实时互动的状态”。

好处

通过数组直接记录值,减少重复递归过程来减少运行时间。

注意

数组下标不能为负数,对于负数直接特判。

宏定义

(第一次接触记录下用法)
(#define) (定义的对象)(定义的含义)

#include <iostream>
#include <string>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
ll d[30][30][30];

int w(ll a, ll b, ll c) {

	if (a <= 0 || b <= 0 || c <= 0) {


		return 1;
	} else if (a > 20 || b > 20 || c > 20) {
		if (d[20][20][20] == 0) {
			d[20][20][20] = w(20, 20, 20);
		}
		return d[20][20][20];
	} else if (a < b & b < c) {
		if (d[a][b][c] == 0) {
			d[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
		}
		return d[a][b][c];
	} else {
		if (d[a][b][c] == 0) {
			d[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 d[a][b][c];
	}
}

int main() {
	while (true) {
		ll a, b, c;
		cin >> a >> b >> c;
		if (a == -1 & b == -1 & c == -1) {
			break;
		} else {
			cout << "w(" << a << ", " << b << ", " << c << ") = " << w(a, b, c) << endl;
		}
	}
	return 0;
}