算法入门复习4

洛谷刷题记录
【算法1-4】递推与递归
P1464 Function
原题目地址：https://www.luogu.com.cn/problem/P1464

题目说了一个递归函数，但是这个递归函数在数字较大时会花费非常多的时间，并且其中很多的情况都是重复运算。

因此第一步先把这个题目模拟出来，在模拟的时候我犯了一个错，其中题目说了是a，b，c其中一个大于20则返回w（20， 20， 20），而我在写的时候不小心写成了 >= 导致了无限循环，我还没发现 模拟完之后进行记忆化搜索，只需要把已经运算过的结果储存起来就好了，因为a，b，c只要有一个超过20就是w（20， 20， 20），所以只需要开一个三维数组，m[21][21][21]
每一次运行的时候去储存，后面调用函数的时候都去看一下这个有没有运算过，运算过则直接使用即可。

然后果不其然的WA了 在观察了一会之后发现是因为a，b，c的数字可能很大，而我的代码是这样写的：

	else if (m[a][b][c] != 0) return m[a][b][c];
	
	else if (a > 20 || b > 20 || c > 20){
		return w(20, 20, 20);
	}

在判断a，b，c之前就进行了查看，如果这时候a，b，c其中有一个超过21则超出了数组的范围，导致RE，最后调换了这两个if的位置，成功过关。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

int m[21][21][21];

ll w(ll a, ll b, ll c){
	
	if(a <= 0 || b <= 0 || c <= 0) return 1;

	else if (a > 20 || b > 20 || c > 20){
		return w(20, 20, 20);
	}
	
	else if (m[a][b][c] != 0) return m[a][b][c];
	

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

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