2021.06.01单词接龙

2021.06.01单词接龙

(题目来源：https://www.luogu.com.cn/problem/P1019 )

题目描述

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

输入描述

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

输出描述

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

样例输入

5
at
touch
cheat
choose
tact
a

样例输出

23

数据规模和约定

n <= 20

思路

1. 由于数据量较小，可以直接dfs遍历。
2. int数组记录状态：一个字符串的使用次数。

代码

    class Solution{ 
        String[] strs;
        //由于交接处的最大长度不能超过字符串的最大长度，所以每次只需要匹配[1,maxLen-1]长度的字串
        int maxLen = -1; 
        int n, res= 0;
        void test() {
            Scanner cin = new Scanner(System.in);
            n = cin.nextInt();
            cin.nextLine();
            strs = new String[n];
            for(int i = 0; i < n; i++) {
                strs[i] = cin.nextLine();
                maxLen = Math.max(maxLen, strs[i].length());
            }
            String head = cin.nextLine();
            //********************* i/o ************************
            dfs(new int[n], head, 1);
            System.out.println(res);
        }
        //访问次数， 当前下标， 用来比较的字符串（最多只需要记录maxLen长度） 总串长度
        void dfs(int[] vis, String s, int len) {
            res = Math.max(res, len);
            for(int l = 1; l <= Math.min(s.length(), maxLen-1); l++) { //用来比较的长度
                char[] sub = s.substring(s.length()-l).toCharArray();
                for(int i = 0; i < n; i++) { //看第i个字符串是否可以接龙
                    if(vis[i] >= 2) continue;
                    boolean can = true;
                    for(int j = 0; j < Math.min(l, strs[i].length()); j++) {
                        if(sub[j] != strs[i].charAt(j)) {
                            can = false;
                            break;
                        }
                    }
                    if(can) { //如果可以
                        StringBuffer sb = new StringBuffer(s);
                        for(int k = l; k < strs[i].length(); k++) { //连接字符串
                            sb.append(strs[i].charAt(k));
                        }
                        vis[i]++;
                        dfs(vis, sb.toString().substring(Math.max(0, sb.length()-maxLen)), len+strs[i].length()-l);
                        vis[i]--; //状态回溯
                    }
                }
            }
        }
    }