洛谷P1019 单词接龙

洛谷P1019 单词接龙

题目描述

单词接龙是一个与我们经常玩的成语接龙相类似的游戏，现在我们已知一组单词，且给定一个开头的字母，要求出以这个字母开头的最长的“龙”（每个单词都最多在“龙”中出现两次），在两个单词相连时，其重合部分合为一部分，例如 beastbeast和astonishastonish，如果接成一条龙则变为beastonishbeastonish，另外相邻的两部分不能存在包含关系，例如atat 和 atideatide 间不能相连。

输入输出格式

输入格式：
输入的第一行为一个单独的整数nn (n \le 20n≤20)表示单词数，以下nn 行每行有一个单词，输入的最后一行为一个单个字符，表示“龙”开头的字母。你可以假定以此字母开头的“龙”一定存在.

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

样例

输入：

5
at
touch
cheat
choose
tact
a

输出：

23

代码解析：
难点在于龙的衔接部分 ，使用预处理各个单词之间的关系之后就能简化DFS的要求。

#include<bits/stdc++.h>
using namespace std;
string word[25];
int book[25];
int ans = 0;
int arr[25][25]; //预处理后的数组
int n;
int check(int a , int b){
//a与b相连 ，寻找b的能够和a相连的位置并返回 ， 如果找不到就返回0
	int al = word[a].length();
	int bl = word[b].length();
	for(int k = 1; k < min(al , bl); k ++){
		int flag = 1;
		int i = al - k , j = 0;
		while(i < al && j < k){
			if(word[a][i] != word[b][j])
			{
				flag = 0;
				break;
			}
			i ++ , j ++;
		}
		if(flag) return k;
	}
	return 0;
}
void dfs(int pre , int len){
	ans = max(ans , len);
	for(int i = 0; i < n; i ++){
		if(arr[pre][i] == 0 || book[i] > 1) continue;
		book[i] ++;
		dfs(i , len + word[i].length() - arr[pre][i]);
		book[i] --;
	}
}
int main(){
	cin >> n;
	char start;
	for(int i = 0; i < n; i ++){
		cin >> word[i];
	}

	cin >> start;
	for(int i = 0; i < n; i ++){
		for(int j = 0; j < n; j ++){
			arr[i][j] = check(i , j);
		}
	}
	for(int i = 0; i < n; i ++){
	//遍历每个可能性
		if(word[i][0] == start)
		{
			book[i] ++;
			dfs(i , word[i].length());
			book[i] --;
		}
	}
	cout << ans;
}

引用自：原文连接