【字典树】745. 前缀和后缀搜索

题目：

题解

维护两颗字典树，一个是（前缀）树t1，另一个是（后缀）树t2。

定义字典树的节点，假设已经理解字典树的存储方式，此处应该在字典树的节点上添加一个属性List<Integer> idxs，意为数组words中拥有此前缀的节点的字符串的下标。

所以字典树的节点为：

class Trie {
     Trie[] children = new Trie[26];
     List<Integer> idxs = new ArrayList<>();
 }

把words分别存储在t1与t2中，在添加时，将当前word的下标i存入Trie的idxs属性中

查询时，按照前缀查询t1，得到下标集合l1，按照后缀查询t2，得到下标集合l2

因为l1和l2都是有序的，题目要求取最大的，所以按照双指针来取出最大下标即为结果

code

class WordFilter {

    Trie t1 = new Trie(), t2 = new Trie();

    public WordFilter(String[] words) {
        int n = words.length;
        for (int i = 0; i < n; i++) {
            add(t1, words[i], i, false);
            add(t2, words[i], i, true);
        }
    }
    
    public int f(String pref, String suff) {
        return query(pref, suff);
    }

    private int query(String a, String b) {
        int n = a.length(), m = b.length();
        Trie t = t1;
        for (int i = 0; i < n; i++) {
            int p = a.charAt(i) - 'a';
            if (t.children[p] == null) return -1;
            t = t.children[p];
        }
        List<Integer> l1 = t.idxs;
        t = t2;
        for (int i = m - 1; i >= 0; i--) {
            int p = b.charAt(i) - 'a';
            if (t.children[p] == null) return -1;
            t = t.children[p];
        }
        List<Integer> l2 = t.idxs;
        n = l1.size(); m = l2.size();
        for (int i = n - 1, j = m - 1; i >= 0 && j >= 0; ) {
            if (l1.get(i) > l2.get(j)) i--;
            else if (l1.get(i) < l2.get(j)) j--;
            else return l1.get(i);
        }
        return -1;
    }

    private void add(Trie t, String s, int idx, boolean isTurn) {
        int n = s.length();
        t.idxs.add(idx);
        for (int i = isTurn ? n - 1 : 0; i >= 0 && i < n; i += isTurn ? -1 : 1) {
            int p = s.charAt(i) - 'a';
            if (t.children[p] == null) t.children[p] = new Trie();
            t = t.children[p];
            t.idxs.add(idx);
        }
    }

    class Trie {
        Trie[] children = new Trie[26];
        List<Integer> idxs = new ArrayList<>();
    }

}

• 时间复杂度：初始化操作复杂度为$O({\sum }_{i=0}^{n-1}ss[i].length)$，检索的时间复杂度为$O(a+b+n)$，其中$a=b=7$为前后缀最大长度，$n=1e4$ 为初始化数组长度，代表最多有n个候选下表

• 空间复杂度：$O({\sum }_{i=0}^{n-1}ss[i].length)$

ps：
本篇题解主要是来自于宫水三叶的解析，真的很强，自己尝试写下，关于时间复杂度和空间复杂度的公式写法，还是有待学习的