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

来源：互联网发布：公路计算软件编辑：程序博客网时间：2024/06/05 02:54

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;}

0 0