【DFS（记忆化）】poj 1351 Number of Locks

http://poj.org/problem?id=1351

题意：有一个有4种高度{1,2,3,4}，n个卡槽的锁，要求该锁卡槽：1）至少有三种不同的高度；2）至少有一对相邻的两个卡槽的高度差为1（即：存在1、4相邻），求有多少种可能？

分析：四种状态：dp[t][pre][f][dif]

t：第t个卡槽；pre：前一个卡槽；f：是否存在高度差为3 的卡槽对；dif：当前使用了几种卡槽

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;

__int64 dp[25][5][2][5];
int vis[5],n;

__int64 DFS(int t,int pre,int f,int dif)
{
	if(dp[t][pre][f][dif]) return dp[t][pre][f][dif];
	if(t>=n){
		if(f&&dif>=3) return 1;
		else return 0;
	}
	for(int i=1;i<=4;i++){
		if(!vis[i]){
			vis[i]=1;  //添加了新的卡槽
			dp[t][pre][f][dif]+=DFS(t+1,i,f||(pre&&abs(pre-i)==3),dif+1);
			vis[i]=0;
		}
		else  dp[t][pre][f][dif]+=DFS(t+1,i,f||(pre&&abs(pre-i)==3),dif);
	}
	return dp[t][pre][f][dif];
}

int main()
{
	while(scanf("%d",&n)){
		if(n==-1) break;
		memset(dp,0,sizeof(dp));
		memset(vis,0,sizeof(vis));
		printf("%d: %I64d\n",n,DFS(0,0,0,0));
	}
	return 0;
}