POJ 1597 Function Run Fun

题目描述

记忆化搜索典型例题

本题是简单的一个递归函数，但是只是按照题目意思来写一定会TLE
需要创建一个三维数组存储结果，采取记忆化搜索才能AC

#include<iostream>
#include<cstring>
using namespace std;
int f[21][21][21];
int w(int a,int b,int c){
	if(a<=0||b<=0||c<=0)
	return 1;
	else if(a>20||b>20||c>20)
	return w(20,20,20);
	else if(f[a][b][c]) return f[a][b][c];
	else if(a<b&&b<c) return f[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
	else return f[a][b][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);
}
int main(){
	memset(f,0,sizeof(f));
	int a,b,c;
	while(cin>>a>>b>>c){
		if(a==-1&&b==-1&&c==-1)
		return 0;
		cout<<"w("<<a<<", "<<b<<", "<<c<<") = "<<w(a,b,c)<<endl;
	}
}