P1464 Function-记忆化搜索

P1464 Function - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
 

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
int f[30][30][30];//记忆化数组
int dfs(int a,int b,int c){
	if(a<=0||b<=0||c<=0) return 1;
	else if(a>20||b>20||c>20) return dfs(20,20,20);
	else if(f[a][b][c]!=0) return f[a][b][c];//如果该值计算过了，则直接用计算过的。一定要在前两个特判之后，因为f数组的范围不能越界
	else if(a<b&&b<c) f[a][b][c]=dfs(a,b,c-1)+dfs(a,b-1,c-1)-dfs(a,b-1,c);
	else f[a][b][c]=dfs(a-1,b,c)+dfs(a-1,b-1,c)+dfs(a-1,b,c-1)-dfs(a-1,b-1,c-1);
	return f[a][b][c];
}
void solve(){
	while(true){
		int a,b,c;
		cin>>a>>b>>c;
		if(a==-1&&b==-1&&c==-1) break;
		cout<<"w("<<a<<", "<<b<<", "<<c<<") = "<<dfs(a,b,c)<<endl;
	}
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	solve();
	return 0;
}