Description
给定一个函数 f(a, b, c):
如果 a ≤ 0 或 b ≤ 0 或 c ≤ 0 返回值为 1;
如果 a > 20 或 b > 20 或 c > 20 返回值为 f(20, 20, 20);
如果 a < b 并且 b < c 返回 f(a, b, c−1) + f(a, b−1, c−1) − f(a, b−1, c);
其它情况返回 f(a−1, b, c) + f(a−1, b−1, c) + f(a−1, b, c−1) − f(a-1, b-1, c-1)。
看起来简单的一个函数?你能做对吗?
Input
输入包含多组测试数据,对于每组测试数据:
输入只有一行为 3 个整数a, b, c(a, b, c < 30)。
Output
对于每组测试数据,输出函数的计算结果。
Sample Input
1 1 1 2 2 2
Sample Output
2 4
这道题给出了递归的方程,但是直接照抄的话会超时,计算量太大,要使用使用记忆化搜索;
思路如下:
1.利用三维数组存储递归产生的结果,如果以后遇到相同递归,直接调用;
2.在每次递归之前加入判断,判断之前有没有存储过递归结果,存储过的话直接调用;
代码如下:
#include<bits/stdc++.h>
using namespace std;
int f[35][35][35]={{0},{0},{0}}; //建立数组存储结果
int m(int a,int b,int c)
{
if(f[a][b][c]>0)
return f[a][b][c]; //判断是否存储过
else
{
if(a<=0||b<=0||c<=0)
return 1;
else if(a<b&&b<c)
{
f[a][b][c]=(m(a,b,c-1)+m(a,b-1,c-1)-m(a,b-1,c));
return f[a][b][c];
}
else
{
f[a][b][c]=(m(a-1,b,c)+m(a-1,b-1,c)+m(a-1,b,c-1)-m(a-1,b-1,c-1));
return f[a][b][c];
}
}
}
int main()
{
int a,b,c;
while(cin>>a>>b>>c)
{
if(a<=0||b<=0||c<=0)
a=b=c=0;
else if(a>20||b>20||c>20) //这里直接除去了题目中的两个要求,就不用递归了
a=b=c=20;
cout<<m(a,b,c)<<endl;
}
}