【洛谷1464 Function】记忆化搜索

题目链接：传送门

一开始还想着这么大的数据范围得用map吧，后来才发现自己傻了，他的条件筛选已经把范围控制在0-20内了，直接上三维数组记忆化，就很niceeeeeeeee。深刻理解一下记忆化搜索的本质。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
#define inf 4000000000000000000
using namespace std;
typedef long long ll;
const ll maxn=1e6+50;
const ll mod=1e9+7;
const double eps=1e-9;
ll mp[50][50][50];

ll w(ll a,ll b,ll c)
{
	if(a<=0||b<=0||c<=0)
	{
		return 1;
	}
	else if(a>20||b>20||c>20)
	{
		ll tmp=w(20,20,20);
		return tmp;
	}
	else if(a<b&&b<c)
	{
		if(mp[a][b][c]!=inf)
		{
			return mp[a][b][c];
		}
		ll tmp1=w(a,b,c-1);
		ll tmp2=w(a,b-1,c-1);
		ll tmp3=w(a,b-1,c);
		mp[a][b][c]=tmp1+tmp2-tmp3;
		return tmp1+tmp2-tmp3;
	}
	else
	{
		if(mp[a][b][c]!=inf)
		{
			return mp[a][b][c];
		}
		ll t1=w(a-1,b,c);
		ll t2=w(a-1,b-1,c);
		ll t3=w(a-1,b,c-1);
		ll t4=w(a-1,b-1,c-1);
		mp[a][b][c]=t1+t2+t3-t4;
		return t1+t2+t3-t4;
	}
}

int main()
{
	ll a,b,c;
	for(ll i=0;i<=21;i++)
	{
		for(ll j=0;j<=21;j++)
		{
			for(ll k=0;k<=21;k++)
			{
				mp[i][j][k]=4e18;
			}
		}
	}
	while(scanf("%lld %lld %lld",&a,&b,&c)!=EOF)
	{
		if(a==-1&&b==-1&&c==-1)return 0;
		printf("w(%lld, %lld, %lld) = %lld\n",a,b,c,w(a,b,c));
	}
	return 0;
}

感受一下AC的快乐