欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

单词接龙 DFS

程序员文章站 2022-03-11 11:57:48
原题链接单词接龙是一个与我们经常玩的成语接龙相类似的游戏。现在我们已知一组单词,且给定一个开头的字母,要求出以这个字母开头的最长的“龙”,每个单词最多被使用两次。在两个单词相连时,其重合部分合为一部分,例如 beast 和 astonish ,如果接成一条龙则变为 beastonish。我们可以任意选择重合部分的长度,但其长度必须大于等于1,且严格小于两个串的长度,例如 at 和 atide 间不能相连。输入格式输入的第一行为一个单独的整数 n 表示单词数,以下 n 行每行有一个单词(只含有大写...

原题链接

单词接龙是一个与我们经常玩的成语接龙相类似的游戏。

现在我们已知一组单词,且给定一个开头的字母,要求出以这个字母开头的最长的“龙”,每个单词最多被使用两次。

在两个单词相连时,其重合部分合为一部分,例如 beast 和 astonish ,如果接成一条龙则变为 beastonish。

我们可以任意选择重合部分的长度,但其长度必须大于等于1,且严格小于两个串的长度,例如 at 和 atide 间不能相连。

输入格式
输入的第一行为一个单独的整数 n 表示单词数,以下 n 行每行有一个单词(只含有大写或小写字母,长度不超过20),输入的最后一行为一个单个字符,表示“龙”开头的字母。

你可以假定以此字母开头的“龙”一定存在。

输出格式
只需输出以此字母开头的最长的“龙”的长度。

数据范围
n≤20
输入样例:
5
at
touch
cheat
choose
tact
a
输出样例:
23
提示
连成的“龙”为 atoucheatactactouchoose。

一道DFS的题目,要注意的是状态是一整个整体,所以要回溯!

首先计算出,每个单词前后缀重叠的最小单词数,然后dfs一遍开头的单词

#include<iostream>
#include<string>
using namespace std;

const int N = 25;
int n,ans,used[N],con[N][N];
string a[N];
char head;

void set() //计算每个单词前后缀重叠最小单词数,储存在con数组中
{
	for(int i = 0; i < n; i++)
	{
		for(int j = 0; j < n; j++)
		{
			int len =  min(a[i].size(),a[j].size());
			for(int k = 1; k < len; k++)
			{
				if(a[i].substr(a[i].size() - k) == a[j].substr(0,k))
				{
					con[i][j] = k;
					break;
				}	
			}
		}
	}
}

int dfs(int cur,int len)//cur 当前单词下标,len 当前拼接的长度
{
	ans = max(ans,len);//更新答案
	used[cur]++; //使用次数
	
	for(int i = 0; i < n; i++)
	{
		if(used[i] >= 2)	continue;
		//超过两次
		if(con[cur][i])//能拼接上
		{
			dfs(i,len + a[i].size() - con[cur][i]);//递归
			used[i]--;//回溯
		}
	}
}

int main()
{
	cin >> n;
	for(int i = 0; i < n; i++)	cin >> a[i];
	cin >> head;
	
	set();	
	
	for(int i = 0; i < n; i++)
		if(a[i][0] == head)
			dfs(i,a[i].size());
	
	cout << ans << endl;
	
	return 0;
}

本文地址:https://blog.csdn.net/yulong_D/article/details/107881292