1048. 最长字符串链

转载请声明地址四元君

1048. 最长字符串链

题目难度 Medium

给出一个单词列表，其中每个单词都由小写英文字母组成。

如果我们可以在 word1 的任何地方添加一个字母使其变成 word2，那么我们认为 word1 是 word2 的前身。例如，“abc” 是 “abac” 的前身。

词链是单词 [word_1, word_2, …, word_k] 组成的序列，k >= 1，其中 word_1 是 word_2 的前身，word_2 是 word_3 的前身，依此类推。

从给定单词列表 words 中选择单词组成词链，返回词链的最长可能长度。

示例：

输入：["a","b","ba","bca","bda","bdca"]
输出：4
解释：最长单词链之一为 "a","ba","bda","bdca"。

提示：

1 <= words.length <= 1000
1 <= words[i].length <= 16
words[i] 仅由小写英文字母组成。

思路

之前做过的一道题是找到words里面可以凑成的最大字符串。那个做法可以是哈希表或者是前缀树。最好是前缀树。

这个题一时间没想到前缀树具体该怎么做，用了稍微暴力的方法。遍历每一个字符串，看从这个字符串出发得到的最大值是多少，如果满足条件，就深度搜索。使用一个全局的map记录该位置的字符串可有的最大词链长度。如果走过了就直接返回。最后记录这个最大值返回。

代码

class Solution {
public:
    map<string,int> dp;
    bool isMade(string a,string b){
        int flag = 0;
        for(int i=0,j=0;i<a.size()&&j<b.size();){
            if(a[i]!=b[j]){
                if(flag==0){
                    flag = 1;
                    i++;
                }
                else return false;
            }else{
                i++,j++;
            }
        }
        return true;
    }

    int calLen(string &s,vector<string>& words){
        if(dp.count(s))return dp[s];
        int ans = 1;
        for(int i=0;i<words.size();i++){
            if(words[i].size()-1==s.size()&&isMade(words[i],s)){
                ans = max(ans,calLen(words[i],words)+1);
            }
        }
        return dp[s]=ans;
    }

    int longestStrChain(vector<string>& words) {
        int ans = 1;
        for(int i=0;i<words.size();i++){
            ans = max(ans,calLen(words[i],words));

        }
        return ans;
    }
};