洛谷 [P1464 Function] {递归与递推+宏定义+记忆化} 奋斗的珂珂~

题目描述

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

如果a≤0 or b≤0 or c≤0就返回值1.
如果a>20 or b>20 or 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)
这是个简单的递归函数，但实现起来可能会有些问题。当a,b,c均为15时，调用的次数将非常的多。你要想个办法才行.

absi2011 : 比如 w(30,−1,0)既满足条件1又满足条件2 这种时候我们就按最上面的条件来算 所以答案为1

输入格式

会有若干行。

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

保证输入的数在[-9223372036854775808,9223372036854775807][−9223372036854775808,9223372036854775807]之间，并且是整数。

输出格式

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

w(a, b, c) = ans

注意空格。

输入输出样例

输入 #1复制
1 1 1
2 2 2
-1 -1 -1
输出 #1复制
w(1, 1, 1) = 2
w(2, 2, 2) = 4

说明/提示

记忆化搜索

解题思路

题目中已经明显的说明需要记忆化进行处理，这道题使用宏定义可以省去很多的代码。使用wfine来表明是否进行处理过。wfine数组只需要开到大于等于20就可以。

注意

输出格式，空格位置！！！

完整代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll; 
#define wfine(a,b,c) (wfine[a][b][c]?wfine[a][b][c]:wfine[a][b][c]=w(a,b,c))
int wfine[21][21][21];


ll w(ll a,ll b,ll c)
{
	if(a<=0||b<=0||c<=0) return 1;//数组下标不可能为负数，所以直接返回就可以 
	if(a>20||b>20||c>20) return wfine(20,20,20);
	if(a<b&&b<c) return wfine(a,b,c-1)+wfine(a,b-1,c-1)-wfine(a,b-1,c);
	return wfine(a-1,b,c)+wfine(a-1,b-1,c)+wfine(a-1,b,c-1)-wfine(a-1,b-1,c-1);
}
int main()
{
	ll a,b,c;	
	while(1)
	{ 
	    scanf("%lld %lld %lld",&a,&b,&c); 
	    if(a==-1&&b==-1&&c==-1)
	     break;
		printf("w(%lld, %lld, %lld) = %lld\n",a,b,c,w(a,b,c));
	}
	return 0;
 }