洛谷1464 -- Function

题目描述 Description

对于一个递归函数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时，调用的次数将非常的多。你要想个办法才行.

输入输出格式 Input/output

输入格式：

会有若干行.

并以-1，-1，-1结束.

输出格式：

输出若干行

输入输出样例 Sample input/output

样例测试点#1

输入样例： 在线IDE

输出样例：

说明 description

记忆化搜索

用三维数组保存计算过的值

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
int done[22][22][22];
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);
	}
	else if(a < b && b < c)
	{
		if(done[a][b][c])
		{
			return done[a][b][c];
		}
		done[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
		return w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c);
	}
	else
	{
		if(done[a][b][c])
		{
			return done[a][b][c];
		}
		done[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 w(a - 1, b, c) + w(a - 1, b - 1, c) + w(a - 1, b, c - 1) - w(a - 1, b - 1, c - 1);
	}

}
void hello()
{
	int a = 0, b = 0, c = 0;
	cin>>a>>b>>c;
	while(true)
	{
		if(-1 == a && -1 == b && -1 == c)
		{
			return;
		}
		cout<<"w("<<a<<" ,"<<b<<" ,"<<c<<")"<<" = "<<w(a, b, c)<<endl;
		printf("w(%d, %d, %d) = %d\n",a,b,c,w(a,b,c));
		cin>>a>>b>>c;
	}
}
int main()
{
	while(true)
	{
		hello();
	}
	return 0;
}