LeetCode_前缀树_中等_1268.搜索推荐系统

1.题目

给你一个产品数组 products 和一个字符串 searchWord ，products 数组中每个产品都是一个字符串。

请你设计一个推荐系统，在依次输入单词 searchWord 的每一个字母后，推荐 products 数组中前缀与 searchWord 相同的最多三个产品。如果前缀相同的可推荐产品超过三个，请按字典序返回最小的三个。

请你以二维列表的形式，返回在输入 searchWord 每个字母后相应的推荐产品的列表。

示例 1：
输入：products = [“mobile”,“mouse”,“moneypot”,“monitor”,“mousepad”], searchWord = “mouse”
输出：[
[“mobile”,“moneypot”,“monitor”],
[“mobile”,“moneypot”,“monitor”],
[“mouse”,“mousepad”],
[“mouse”,“mousepad”],
[“mouse”,“mousepad”]
]
解释：按字典序排序后的产品列表是 [“mobile”,“moneypot”,“monitor”,“mouse”,“mousepad”]
输入 m 和 mo，由于所有产品的前缀都相同，所以系统返回字典序最小的三个产品 [“mobile”,“moneypot”,“monitor”]
输入 mou， mous 和 mouse 后系统都返回 [“mouse”,“mousepad”]

示例 2：
输入：products = [“havana”], searchWord = “havana”
输出：[[“havana”],[“havana”],[“havana”],[“havana”],[“havana”],[“havana”]]

示例 3：
输入：products = [“bags”,“baggage”,“banner”,“box”,“cloths”], searchWord = “bags”
输出：[[“baggage”,“bags”,“banner”],[“baggage”,“bags”,“banner”],[“baggage”,“bags”],[“bags”]]

示例 4：
输入：products = [“havana”], searchWord = “tatiana”
输出：[[],[],[],[],[],[],[]]

提示：
1 <= products.length <= 1000
1 <= Σ products[i].length <= 2 * 10⁴
products[i] 中所有的字符都是小写英文字母。
1 <= searchWord.length <= 1000
searchWord 中所有字符都是小写英文字母。

2.思路

（1）前缀树
思路参考本题官方题解。

3.代码实现（Java）

//思路1————前缀树
class Solution {
    class TrieNode {
        public static final int num = 26;
        TrieNode[] next;
        PriorityQueue<String> queue;

        public TrieNode() {
            next = new TrieNode[num];
            queue = new PriorityQueue<>((o1, o2) -> o2.compareTo(o1));
        }
    }

    private static final int size = 3;
    private TrieNode root;
    private List<List<String>> res;

    public void insert(String word) {
        TrieNode node = root;
        for (char c : word.toCharArray()) {
            int index = c - 'a';
            if (node.next[index] == null) {
                node.next[index] = new TrieNode();
            }
            node = node.next[index];
            node.queue.offer(word);
            if (node.queue.size() > size) {
                node.queue.poll();
            }
        }
    }

    public void startWith(String word) {
        TrieNode node = root;
        boolean exist = true;
        for (char c : word.toCharArray()) {
            int index = c - 'a';
            if (!exist || node.next[index] == null) {
                exist = false;
                res.add(new ArrayList<>());
                continue;
            }
            node = node.next[index];
            List<String> tmp = new ArrayList<>();
            while (!node.queue.isEmpty()) {
                tmp.add(node.queue.poll());
            }
            Collections.reverse(tmp);
            res.add(tmp);
        }
    }

    public List<List<String>> suggestedProducts(String[] products, String searchWord) {
        res = new ArrayList<>();
        root = new TrieNode();
        for (String s : products) {
            insert(s);
        }
        startWith(searchWord);
        return res;
    }
}