POJ1351 Number of Locks 记忆化搜索

第二道记忆化搜索的题目。其实一直都没搞太明白，这道题好像也比较冷门，网上资料也不多，主要有：

http://blog.csdn.net/scorpiocj/article/details/6642177

http://hi.baidu.com/xiangsanzi/blog/item/04aca409596260a92eddd4fb.html

这样两种方法，另外也可以推公式。代码主要参考了第一个链接，还有一些地方不是很清楚。

AC_CODE：

#include <stdio.h>
#include <math.h>
#include <memory.h>

int n;
//一次WA c设的太小了
long long dp[18][5][2][4];

//a用二进制表示 0001表示有一种高度 0010表示有两种高度
long long dfs(int s,int h,int f,int c,int a)
{
	if(s>=n)
	{
		if(f&&c>=3)
			return 1;
		else return 0;
	}
	if(dp[s][h][f][c]!=-1) return dp[s][h][f][c];
	int t;
	long long ans=0;
	for(int i=1;i<=4;i++)
	{
		//如果第i种高度没有设置，如当前有两种高度，
		//则在搜索下一个槽的时候加一种高度
		t=(a&(1<<(i-1)))?c:(c+1);
		t=(t>3)?3:t;
		ans+=dfs(s+1,i,(h&&abs(i-h)==3)||f,t,a|(1<<(i-1)));
	}
	dp[s][h][f][c]=ans;
	return ans;
}
int main()
{
	while(scanf("%d",&n)&&(n!=-1))
	{
		memset(dp,-1,sizeof(dp));
		dfs(0,0,0,0,0);
		printf("%d: %lld\n",n,dp[0][0][0][0]);
	}
	return 0;
}