题解: 首先想到直接用dfs按题目要求搜一遍就行,结果一提交全部TLE,发现应该用记忆化搜索:
记忆化搜索: 就是在搜索过程中可能同一组值得结果第一次就算出来了,第二次搜索就不要在去搜索这个以前就算出来的值,下一次搜到已经保存的值可以直接使用不用继续去搜索求值!可以大大减少搜索的深度和此数,这样就可以全部AC。百度百科:点击这里!
具体解释: 因为是三个变量,即三个参数,要用一个三维数组来记录当前值,将搜索途中出现参数变化后的数组算出值的就将其保存到数组,在以后搜索时先判断一下当前参数对应的数组是不是0,不是0则说明当前这组参数对应的值已知了,直接return回去即可,而不是继续去做重复无用的工作!
参考代码: 记忆化DFS
#include <queue>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
#define MAX 5010
using namespace std;
typedef long long ll;
ll a, b, c, k;
int S[30][30][30];
ll dfs(ll aa, ll bb, ll cc)
{
if(aa <= 0 || bb <= 0 || cc <= 0) return 1;
else if(aa > 20 || bb > 20 || cc > 20) return dfs(20, 20, 20);
if(!S[aa][bb][cc])
{
if(aa < bb && bb < cc) S[aa][bb][cc] = dfs(aa, bb, cc - 1) + dfs(aa, bb - 1, cc - 1) - dfs(aa, bb - 1, cc);
else return S[aa][bb][cc] = dfs(aa - 1, bb, cc) + dfs(aa - 1, bb - 1, cc) + dfs(aa - 1, bb, cc - 1) - dfs(aa - 1, bb - 1, cc - 1);
}
return S[aa][bb][cc];
}
int main()
{
while(1)
{
cin >> a >> b >> c;
if(a == -1 && b == -1 && c == -1) break;
k = dfs(a, b, c); printf("w(%lld, %lld, %lld) = %lld\n", a, b, c, k);
}
return 0;
}