算法训练-单词接龙

资源限制
时间限制：1.0s 内存限制：256.0MB

问题描述

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

输入格式

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

输出格式

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

样例输入
　　5
　　at
　　touch
　　cheat
　　choose
　　tact
　　a
样例输出
23
样例说明
连成的“龙”为atoucheatactactouchoose

我觉得我能解释的都写在注释中了

package lanqiao;
import java.util.*;
public class _单词接龙 {
	public static int dis[][];//dis[i][j]表示arr[i]字符串与arr[j]字符串连接时会重合多少个字符
	public static int max=0;
	public static List<List<Integer>> list;//记录下标为i的字符串后面可以连接哪些下标的字符串
	public static String[] arr;//保存字符串
	public static int vis[];//记录该字符串被使用过几次
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		int n=in.nextInt();
		arr=new String[n];
		dis=new int[n][n];
		vis=new int[n];
		list=new ArrayList();
		in.nextLine();
		for(int i=0;i<n;i++) {
			arr[i]=in.nextLine();
		}
		for(int i=0;i<n;i++) {
			list.add(new ArrayList<>());
		}
		
		//建立邻接表
		for(int i=0;i<arr.length;i++) {
			for(int j=0;j<arr.length;j++) {
				if(check(arr,i,j)) {
					list.get(i).add(j);
				}
			}
		}
		char c=in.nextLine().charAt(0);
		for(int i=0;i<arr.length;i++) {//尝试每个字符串
			if(arr[i].charAt(0)==c) {
				vis[i]++;//在接龙中使用过一次
				dfs(i,arr[i].length());//递归
				vis[i]--;//回溯
			}
		}
		System.out.println(max);
	}
	
	//lastIndex::上一次所接字符串下标
	//length::当前龙长度
	public static void dfs(int lastIndex,int length) {
		max=Math.max(length, max);
		//遍历可以接上arr[lastIndex]的字符串，如果该字符串vis<2，则继续乡下递归
		for(int i=0;i<list.get(lastIndex).size();i++) {
			int next=list.get(lastIndex).get(i);
			if(vis[next]<2) {
				vis[next]++;
				dfs(next,length+arr[next].length()-dis[lastIndex][next]);
				vis[next]--;//回溯
			}
		}
	}
	
	
	//判断下标为index2的字符串是否可以接在下标为index1的字符串后面
	//如果可以在dis中记录重合部分的长度
	public static boolean check(String[] arr,int index1,int index2) {
		String str1=arr[index1];
		String str2=arr[index2];
		for(int i=1;i<str1.length()&&i<str2.length();i++) {
			int j=0;
			for(;j<i;j++) {
				if(str1.charAt(str1.length()-i+j)!=str2.charAt(j)) {
					break;
				}
			}
			if(j==i) {
				dis[index1][index2]=i;
				return true;
			}
		}
		return false;
	}
}