【日常训练】745. 前缀和后缀搜索

题目

设计一个包含一些单词的特殊词典，并能够通过前缀和后缀来检索单词。

实现 WordFilter 类：
WordFilter(string[] words) 使用词典中的单词 words 初始化对象。
f(string pref, string suff) 返回词典中具有前缀 prefix 和后缀 suff 的单词的下标。如果存在不止一个满足要求的下标，返回其中 最大的下标 。如果不存在这样的单词，返回 -1 。

示例：
输入
[“WordFilter”, “f”]
[[[“apple”]], [“a”, “e”]]
输出
[null, 0]
解释
WordFilter wordFilter = new WordFilter([“apple”]);
wordFilter.f(“a”, “e”); // 返回 0 ，因为下标为 0 的单词：前缀 prefix = “a” 且 后缀 suff = “e” 。

提示：
1 <= words.length <= 10⁴
1 <= words[i].length <= 7
1 <= pref.length, suff.length <= 7
words[i]、pref 和 suff 仅由小写英文字母组成
最多对函数 f 执行 10⁴ 次调用

代码

package dayLeetCode;

import java.util.HashMap;
import java.util.Map;

public class dayleetcode745 {
    // 先找出每个单词的前缀后缀组合的可能性
    // 因为words[i]、pref、suff仅有小写字母构成，所以可以通过一个特殊字符将前后缀连接起来，将前后缀通过@链接

    Map<String, Integer> map;
    public dayleetcode745(String[] words) {
        map = new HashMap<>();
        for (int k = 0; k < words.length; k++){
            String word = words[k];

            // 遍历所有前后缀情况
            // i 前缀
            // j 后缀
            for (int i = 1; i <= word.length(); i++){
                for (int j = 1; j <= word.length(); j++){
                    map.put(word.substring(0, i) + "@" + word.substring(word.length() - j), k);
                }
            }
        }
    }

    public int f(String pref, String suff) {
        return map.getOrDefault(pref + "@" + suff,-1);
    }
}