Leetcode 1268：搜索推荐系统（超详细的解法！！！）-CSDN博客

本文链接：https://blog.csdn.net/qq_17550379/article/details/103263207

这是一个关于LeetCode 1268题的详细解法，涉及前缀树（Trie）和搜索算法。文章解释了如何通过优化Trie结构和使用排序技巧来解决搜索推荐系统的问题，旨在推荐与输入字符串前缀匹配的最多三个产品，同时保证字典序最小。还提供了代码实现和改进思路，并分享了参考链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给你一个产品数组 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^4
products[i] 中所有的字符都是小写英文字母。
1 <= searchWord.length <= 1000
searchWord 中所有字符都是小写英文字母。

解题思路

首先很容易想到通过Trie来做这个问题。关于Trie在之前问题Leetcode 208:实现 Trie (前缀树)中出现过了，我们直接在此基础上修改即可。

我们需要求解满足前缀条件的所有字符串，实际上这就是树的遍历，我们需要从Trie和当前匹配的word的最后一个字符出发找到所有的单词，然后将他们排序，返回字典序最小的三个。

class Node:
    def __init__(self, isWord=False):
        self.isWord = isWord
        self.next = collections.defaultdict(Node)
        
class Trie:
    def __init__(self):
        self.root = Node()

    def insert(self, word):
        cur = self.root
        for c in word:
            cur = cur