P1464 Function题解

题目

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

• 如果a≤0或b≤0或c≤0就返回值1。
• 如果a>20或b>20或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时，调用的次数将非常的多。你要想个办法才行。

注意：例如w(30,−1,0)又满足条件1又满足条件2，请按照最上面的条件来算，答案为1。

输入输出格式

输入格式

会有若干行。

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

输出格式

输出若干行，每一行格式：w(a, b, c) = ans，注意空格。

输入输出样例

输入样例

1 1 1
2 2 2
-1 -1 -1

输出样例

w(1, 1, 1) = 2
w(2, 2, 2) = 4

代码

同样使用记忆化搜索。

#include<iostream>
#include<cstdio>
using namespace std;
long long f[25][25][25];
long long w(long long a,long long b,long long c){
	if(a<=0||b<=0||c<=0){
		return 1;
	}
	if(a>20||b>20||c>20){
		return w(20,20,20);
	}
	if(a<b&&b<c){
		if(f[a][b][c-1]==0){
			f[a][b][c-1]=w(a,b,c-1);
		}
		if(f[a][b-1][c-1]==0){
			f[a][b-1][c-1]=w(a,b-1,c-1);
		}
		if(f[a][b-1][c]==0){
			f[a][b-1][c]==w(a,b-1,c);
		}
		f[a][b][c]=f[a][b][c-1]+f[a][b-1][c-1]-f[a][b-1][c];
	}
	else{
		if(f[a-1][b][c]==0){
			f[a-1][b][c]=w(a-1,b,c);
		}
		if(f[a-1][b-1][c]==0){
			f[a-1][b-1][c]=w(a-1,b-1,c);
		}
		if(f[a-1][b][c-1]==0){
			f[a-1][b][c-1]=w(a-1,b,c-1);
		}
		if(f[a-1][b-1][c-1]==0){
			f[a-1][b-1][c-1]=w(a-1,b-1,c-1);
		}
		f[a][b][c]=f[a-1][b][c]+f[a-1][b][c-1]+f[a-1][b-1][c]-f[a-1][b-1][c-1];
	}
	return f[a][b][c];
}
int main(){
	long long a,b,c;
	while(cin>>a>>b>>c){
		if(a==-1&&b==-1&&c==-1){
			return 0;
		}
	    printf("w(%lld, %lld, %lld) = ", a, b, c);
		printf("%lld\n", w(a, b, c));
	}
	return 0;
}