洛谷 1278 记忆化搜索

链接： https://www.luogu.org/problemnew/show/P1278

题目描述

Io和Ao在玩一个单词游戏。

他们轮流说出一个仅包含元音字母的单词，并且后一个单词的第一个字母必须与前一个单词的最后一个字母一致。

游戏可以从任何一个单词开始。

任何单词禁止说两遍，游戏中只能使用给定词典中含有的单词。

游戏的复杂度定义为游戏中所使用的单词长度总和。

编写程序，求出使用一本给定的词典来玩这个游戏所能达到的游戏最大可能复杂度。

输入输出格式

输入格式：

 

输入文件的第一行，表示一个自然数N(1≤N≤16)，N表示一本字典中包含的单词数量以下的每一行包含字典中的一个单词，每一个单词是由字母A、E、I、O和U组成的一个字符串，每个单词的长度将小于等于100，所有的单词是不一样的。

 

输出格式：

 

输出文件仅有一行，表示该游戏的最大可能复杂度。

 

输入输出样例

输入样例#1： 复制

5
IOO
IUUO
AI
OIOOI
AOOI

输出样例#1： 复制

16

 

思路： 直接进行搜索因为重复子状态的问题要被T的，所以在每次搜索完一个状态后记录下这个状态的最优值。下一次再搜到这个状态直接转移就可以了。

代码：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>

using namespace std;

int dp[66000][25];

struct node
{
	char st,en;
	int len;
}dan[25];
int n;
char s[105];
int up;
int yi[25];

void init_yi()
{
	yi[0]=1;
	for(int i=1;i<=21;i++){
		yi[i]=yi[i-1]*2;
	}
}

int jud(int i,int j)
{
	if(dan[i].en==dan[j].st) return 1;
	return 0;
}

int dfs(int s,int pre)
{
	//cout<<"s "<<s<<" pre "<<pre<<endl;
	if(dp[s][pre]!=-1) return dp[s][pre];
	// 先判断一下是否能够搜的动
	int f=0;
	for(int i=0;i<n;i++){
		int t=(1<<i);
		if((s&t)==0&&(jud(pre,i))){
			f=1;break;
		}
	}
	
	if(f==0){
		return dp[s][pre]=0;
	}
	int res=0;
	for(int i=0;i<n;i++){
		int t=yi[i];
		if((s&t)==0&&(jud(pre,i))){
			res=max(res,dfs((s|t),i)+dan[i].len);
		}
	}
	return dp[s][pre]=res;
}

int main()
{
	init_yi();
	memset(dp,-1,sizeof(dp));
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%s",s);
		int len=strlen(s);
		dan[i].st=s[0];
		dan[i].en=s[len-1];
		dan[i].len=len;
	}
	
	
	int ans=0;
	for(int i=0;i<n;i++){
		ans=max(ans,dfs(yi[i],i)+dan[i].len);
	}
	printf("%d\n",ans);
	
	return 0;
}