洛谷 P1464 Function 记忆化搜索

P1464 Function
提交

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

如果a \le 0a≤0 or b \le 0b≤0 or c \le 0c≤0就返回值11.
如果a>20a>20 or b>20b>20 or c>20c>20就返回w(20,20,20)w(20,20,20)
如果a<ba<b并且b<cb<c 就返回w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,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)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,ca,b,c均为15时，调用的次数将非常的多。你要想个办法才行.

/* absi2011 : 比如 w(30,-1,0)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

思路：直接搜会暴，加一个记忆化搜索。

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
ll a,b,c,note[30][30][30]={0};

ll w(ll i,ll j,ll k)  //定义w
{
	if(i<=0||j<=0||k<=0) return 1; //第一种情况
	if(i>20||j>20||k>20) return w(20,20,20); //第二种情况
	if(i<j&&j<k) //第三种情况
	{
		if(note[i][j][k-1]==0) //查询此种情况是否已经存在，没有就储存，以下情况相同；
		{
			note[i][j][k-1]==w(i,j,k-1);
		}
			
		if(note[i][j-1][k-1]==0)
		{
			note[i][j-1][k-1]==w(i,j-1,k-1);
		}
			
		if(note[i][j-1][k]==0)
		{
			note[i][j-1][k]==w(i,j-1,k);
		}
			
		note[i][j][k]=note[i][j][k-1]+note[i][j-1][k-1]-note[i][j-1][k];
	}
	else
	{
		if(note[i-1][j][k] == 0)
		{
			note[i-1][j][k] = w(i-1,j,k);
		}
			
			
		if(note[i-1][j-1][k]==0){
			note[i-1][j-1][k] = w(i-1,j-1,k);
		}
			
			
		if(note[i-1][j][k-1] == 0)
		{
			note[i-1][j][k-1] = w(i-1,j,k-1);
		}
            
            
		if(note[i-1][j-1][k-1]==0)
		{
			note[i-1][j-1][k-1] = w(i-1,j-1,k-1);
		}
			
			
		note[i][j][k]=note[i-1][j][k]+note[i-1][j-1][k]+note[i-1][j][k-1]-note[i-1][j-1][k-1];
	}
	return note[i][j][k];
}

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

}