1048. 最长字符串链

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

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-string-chain
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

记忆化搜索

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class Solution {

    private Map<Integer, List<String>> dictList;

    private Map<String, Integer> dp = new HashMap<>();

    private boolean isNext(String w1, String w2) {
        if (w1.length() + 1 != w2.length()) {
            return false;
        }
        int index1 = 0, index2 = 0;
        while (index1 < w1.length() && index2 < w2.length() && w1.charAt(index1) == w2.charAt(index2)) {
            index1++;
            index2++;
        }
        /**
         * 在末尾添加
         */
        if (index1 == w1.length()) {
            return true;
        }

        index2++;
        while (index1 < w1.length() && index2 < w2.length() && w1.charAt(index1) == w2.charAt(index2)) {
            index1++;
            index2++;
        }
        return index1 == w1.length();
    }

    private int solve(String word) {
        if (!dp.containsKey(word)) {
            int nextLength = word.length() + 1;
            List<String> list = dictList.get(nextLength);
            if (list == null) {
                dp.put(word, 1);
            } else {
                int ans = 0;
                for (String item : list) {
                    if (isNext(word, item)) {
                        ans = Math.max(ans, solve(item));
                    }
                }
                dp.put(word, ans + 1);
            }
        }
        return dp.get(word);
    }

    public int longestStrChain(String[] words) {
        if (words == null || words.length == 0) {
            return 0;
        }

        int minLength = Integer.MAX_VALUE;
        int maxLength = Integer.MIN_VALUE;

        Map<Integer, List<String>> dictList = new HashMap<>();
        for (String word : words) {
            dictList.computeIfAbsent(word.length(), k -> new ArrayList<>()).add(word);
            minLength = Math.min(minLength, word.length());
            maxLength = Math.max(maxLength, word.length());
        }
        this.dictList = dictList;
        int ans = 0;

        for (String word : words) {
            ans = Math.max(ans, solve(word));
        }

        return ans;
    }
}

排序 + 动态规划

import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;

class Solution {
    public int longestStrChain(String[] words) {
        Map<String, Integer> dp = new HashMap<>();
        Arrays.sort(words, Comparator.comparingInt(String::length));
        int res = 0;
        for (String word : words) {
            int best = 0;
            for (int i = 0; i < word.length(); ++i) {
                String prev = word.substring(0, i) + word.substring(i + 1);
                best = Math.max(best, dp.getOrDefault(prev, 0) + 1);
            }
            dp.put(word, best);
            res = Math.max(res, best);
        }
        return res;
    }
}

上一篇:【笔记】并查集---无向图处理代码模板及类型题


下一篇:专访 Christian Posta:Istio 1.7 将成为生产可用的最稳定版本